Update invoice/app/views.py

invoice/app/views.py

@@ -6,46 +6,18 @@ import pandas as pd
 import os
 from dotenv import load_dotenv
 from django.http import JsonResponse
-
+from pymongo import MongoClient
+import json
 load_dotenv()
 
 endpoint = os.environ["endpoint"]
 key = os.environ["key"]
 
-from pymongo import MongoClient
-import json
-
-# Set up MongoDB client
-# client = MongoClient('mongodb://localhost:27017/')
+# client = MongoClient("mongodb://localhost:27017/")
 client = MongoClient('mongodb://db:27017/')
-
-db = client['invoice']  # Database name
-
-def save_to_mongodb(request):
-    if request.method == 'POST':
-        invoice_number = f"invno_{request.POST.get('invoice_number')}"
-        print(invoice_number)
-        tables_data = json.loads(request.POST.get('tables'))
-        print(tables_data)
-
-        # Use the invoice number as the collection name
-        collection = db[invoice_number]
-
-        # Insert each table as a document in MongoDB
-        for table_data in tables_data:
-            headers = table_data['headers']
-            rows = table_data['rows']
-
-            # Create a list of dictionaries for each row
-            table_rows = [dict(zip(headers, row)) for row in rows]
-
-            # Insert into the MongoDB collection
-            collection.insert_many(table_rows)
-            print(collection)
-
-        return JsonResponse({'message': 'Table saved to MongoDB successfully.'})
-
-    return JsonResponse({'error': 'Invalid request'}, status=400)
+db = client["invoices"]
+collection = db["invoice"]
+keyvalue_pair = {}
 
 def upload_document(request):
     context = {'kv_df': [], 'tables': []}
@@ -67,6 +39,7 @@ def upload_document(request):
         result = poller.result()
 
         # Process the result
+        global kv_pairs, tables, keyvalue_pair
         kv_pairs = []
         for kv_pair in result.key_value_pairs:
             if kv_pair.key and kv_pair.value:
@@ -75,7 +48,8 @@ def upload_document(request):
 
         # Convert list of key-value pairs into a format for Django template
         kv_df = [{'key': kv[0], 'value': kv[1]} for kv in kv_pairs]
-
+        keyvalue_pair = {kv[0]:kv[1] for kv in kv_pairs}
+       
         tables = []
         if result.tables:
             for table in result.tables:
@@ -97,6 +71,7 @@ def upload_document(request):
             'file_name': file.name,
             'file_size': file.size,
             'file_type': file.content_type,
+        
         }
 
     return render(request, 'upload.html', context)
@@ -105,3 +80,70 @@ def remove_file(request):
     if request.method == 'POST':
         return JsonResponse({'message': 'File removed successfully.'})
     return JsonResponse({'message': 'Invalid request'}, status=400)
+
+
+# Function to get the field names from MongoDB
+def get_invoice_fields(request):
+    
+    response_data = {
+            'fields': ["Invoice No", "Customer Name", "Invoice Date","Invoice Item", "Quantity", "Base Unit", "Rate", "GST"],  # MongoDB document fields for "Target"
+            'kvPairs': [{'key': key, 'value': value} for key, value in keyvalue_pair.items()]  # Key-value pairs
+        }
+    return JsonResponse(response_data, status=200)
+    
+
+def insert_to_db(request):
+    """
+    This view handles the insertion of mapping data to MongoDB.
+    It expects the data in JSON format where source columns are mapped to target columns.
+    """
+    if request.method == 'POST':
+        try:
+            # Parse the JSON data received from the AJAX request
+            data = json.loads(request.body)
+            mappings = data.get('mappings', [])
+
+            # Dictionary to store all data that will be inserted as new rows (documents) in MongoDB
+            mongo_documents = []
+
+            max_rows = 0
+
+            # First, determine the maximum number of rows based on the selected table columns
+            for mapping in mappings:
+                if 'values' in mapping:
+                    max_rows = max(max_rows, len(mapping['values']))
+
+            # Prepare the documents for MongoDB
+            for i in range(max_rows):
+                # Create a new document for each row
+                document = {}
+
+                # For each mapping, add the column values or key-value pairs to the document
+                for mapping in mappings:
+                    target_field = mapping.get('targetField')
+
+                    if 'values' in mapping:
+                        values = mapping['values']
+                        # Get the value for the current row, or use an empty string if no value exists for this row
+                        value = values[i] if i < len(values) else ''
+                        document[target_field] = value
+                    elif 'value' in mapping:
+                        # Replicate the same key-value pair across all rows
+                        document[target_field] = mapping['value']
+
+                # Add the constructed document to the list
+                mongo_documents.append(document)
+
+            # Insert the documents into MongoDB
+            if mongo_documents:
+                collection.insert_many(mongo_documents)
+
+            return JsonResponse({'status': 'success', 'message': 'Data mapped and saved successfully to MongoDB!'})
+
+        except Exception as e:
+            # Return an error response if something goes wrong
+            return JsonResponse({'status': 'error', 'message': str(e)}, status=500)
+
+    else:
+        # Return an error if the request method is not POST
+        return JsonResponse({'status': 'error', 'message': 'Invalid request method.'}, status=400)