Update app.py

app.py

@@ -150,71 +150,76 @@ def filter_valid_attributes(attributes, valid_fields):
     return {ATTRIBUTE_MAPPING[key]: value for key, value in attributes.items() if ATTRIBUTE_MAPPING[key] in valid_fields}
 
 #📊  Function to interact with Salesforce based on mode and type
-def export_entry_sales(sf, product_data):
-    """
-    Exports data to VENKATA_RAMANA_MOTORS__c and UNBILLING_DATA__c objects when entry and sales are clicked.
-    """
-    venkata_data = {
-        'Product_Name__c': product_data['Product_Name__c'],
-        'Modal_Name__c': product_data['Modal_Name__c'],
-        'Quantity__c': product_data['Quantity']
-    }
-    sf.VENKATA_RAMANA_MOTORS__c.create(venkata_data)
-    
-    unbilling_data = {
-        'Product_Name__c': product_data['Product_Name__c'],
-        'Modal_Name__c': product_data['Modal_Name__c'],
-        'TotalQuantity__c': product_data['Quantity']
+def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
+    try:
+        sf = Salesforce(
+            username=SALESFORCE_USERNAME,
+            password=SALESFORCE_PASSWORD,
+            security_token=SALESFORCE_SECURITY_TOKEN
+        )
+def export_entry_sales(sf, quantity):
+    data = {
+        'Quantity__c': quantity
     }
-    sf.UNBILLING_DATA__c.create(unbilling_data)
-    
-    print("Entry and Sales data exported successfully.")
+    sf.VENKATA_RAMANA_MOTORS__c.create(data)
+    print("Exported to VENKATA_RAMANA_MOTORS__c successfully")
 
-def update_exit_sales(sf, product_data):
-    """
-    Updates Inventory_Management__c and Un_Billable__c objects when exit mode sales is clicked.
-    """
-    # Find matching record in Inventory_Management__c
-    inventory_records = sf.query(f"""
-        SELECT Id, Quantity_Sold__c FROM Inventory_Management__c 
-        WHERE Product_Name__c = '{product_data['Product_Name__c']}' 
-        OR Modal_Name__c = '{product_data['Modal_Name__c']}'
-    """)
-    
-    if inventory_records['records']:
-        inventory_id = inventory_records['records'][0]['Id']
-        new_quantity_sold = inventory_records['records'][0]['Quantity_Sold__c'] + product_data['Quantity']
-        sf.Inventory_Management__c.update(inventory_id, {'Quantity_Sold__c': new_quantity_sold})
-        print("Inventory Management updated successfully.")
-    
-    # Find matching record in Un_Billable__c
-    unbillable_records = sf.query(f"""
-        SELECT Id, Sold_Out__c FROM Un_Billable__c 
-        WHERE Product_Name__c = '{product_data['Product_Name__c']}' 
-        OR Model_Name__c = '{product_data['Modal_Name__c']}'
-    """)
-    
-    if unbillable_records['records']:
-        unbillable_id = unbillable_records['records'][0]['Id']
-        new_sold_out_quantity = unbillable_records['records'][0]['Sold_Out__c'] + product_data['Quantity']
-        sf.Un_Billable__c.update(unbillable_id, {'Sold_Out__c': new_sold_out_quantity})
-        print("Un_Billable updated successfully.")
-
-# Example usage
-sf = Salesforce(username='your_username', password='your_password', security_token='your_security_token')
-
-product_data = {
-    'Product_Name__c': 'Example Product',
-    'Modal_Name__c': 'Example Model',
-    'Quantity': 10
-}
+def export_entry_nonsales(sf, quantity):
+    data = {
+        'TotalQuantity__c': quantity
+    }
+    sf.UNBILLING_DATA__c.create(data)
+    print("Exported to UNBILLING_DATA__c successfully")
 
-# When clicking entry and sales
-export_entry_sales(sf, product_data)
+def export_exit_sales(sf, product_name, quantity):
+    query = f"SELECT Id, Quantity_Sold__c FROM Inventory_Management__c WHERE Name = '{product_name}' LIMIT 1"
+    result = sf.query(query)
+    if result['records']:
+        record_id = result['records'][0]['Id']
+        updated_quantity = result['records'][0]['Quantity_Sold__c'] + int(quantity)
+        sf.Inventory_Management__c.update(record_id, {'Quantity_Sold__c': updated_quantity})
+        print("Updated Inventory_Management__c successfully")
+    else:
+        data = {
+            'Name': product_name,
+            'Quantity_Sold__c': quantity
+        }
+        sf.Inventory_Management__c.create(data)
+        print("Created new record in Inventory_Management__c successfully")
 
-# When clicking exit mode sales
-update_exit_sales(sf, product_data)
+def export_exit_nonsales(sf, product_name, quantity):
+    query = f"SELECT Id, Sold_Out__c FROM Un_Billable__c WHERE Name = '{product_name}' LIMIT 1"
+    result = sf.query(query)
+    if result['records']:
+        record_id = result['records'][0]['Id']
+        updated_quantity = result['records'][0]['Sold_Out__c'] + int(quantity)
+        sf.Un_Billable__c.update(record_id, {'Sold_Out__c': updated_quantity})
+        print("Updated Un_Billable__c successfully")
+    else:
+        data = {
+            'Name': product_name,
+            'Sold_Out__c': quantity
+        }
+        sf.Un_Billable__c.create(data)
+        print("Created new record in Un_Billable__c successfully")
 
+def main():
+    sf = connect_to_salesforce()
+    
+    action = input("Enter action (entry_sales, entry_nonsales, exit_sales, exit_nonsales): ")
+    product_name = input("Enter product/model name: ")
+    quantity = input("Enter quantity: ")
+    
+    if action == "entry_sales":
+        export_entry_sales(sf, quantity)
+    elif action == "entry_nonsales":
+        export_entry_nonsales(sf, quantity)
+    elif action == "exit_sales":
+        export_exit_sales(sf, product_name, quantity)
+    elif action == "exit_nonsales":
+        export_exit_nonsales(sf, product_name, quantity)
+    else:
+        print("Invalid action")
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():
     try: