pavansuresh commited on
Commit
88d3ede
·
verified ·
1 Parent(s): f4137db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -6,6 +6,7 @@ from utils import (
6
  extract_text_from_pdf, extract_key_value_pairs, map_fields,
7
  create_record, attach_pdf, log_failure
8
  )
 
9
 
10
  # Ensure uploads directory exists
11
  os.makedirs("uploads", exist_ok=True)
@@ -116,9 +117,21 @@ def migrate_to_salesforce(pdf_paths, object_name):
116
  if error:
117
  log_failure(pdf_path, object_name, error)
118
  return f"Error: {error}"
119
- data = {mappings[key]: value for key, value in zip(extracted_data[0]["keys"], extracted_data[0]["values"])}
120
 
121
- # Inserted your valid AccountId (Option 2)
 
 
 
 
 
 
 
 
 
 
 
 
 
122
  data["AccountId"] = "001dL00001ASyPbQAL"
123
 
124
  record_id, error = create_record(sf, object_name, data)
 
6
  extract_text_from_pdf, extract_key_value_pairs, map_fields,
7
  create_record, attach_pdf, log_failure
8
  )
9
+ from datetime import datetime # ✅ Required for date conversion
10
 
11
  # Ensure uploads directory exists
12
  os.makedirs("uploads", exist_ok=True)
 
117
  if error:
118
  log_failure(pdf_path, object_name, error)
119
  return f"Error: {error}"
 
120
 
121
+ data = {}
122
+ for key, value in zip(extracted_data[0]["keys"], extracted_data[0]["values"]):
123
+ field_api = mappings.get(key)
124
+ if field_api:
125
+ # ✅ Convert date if value is in MM/DD/YY or M/D/YY format
126
+ try:
127
+ if "/" in value and len(value.split("/")[-1]) == 2:
128
+ parsed_date = datetime.strptime(value.strip(), "%m/%d/%y")
129
+ value = parsed_date.strftime("%Y-%m-%d")
130
+ except Exception:
131
+ pass # If it fails, just keep original
132
+ data[field_api] = value
133
+
134
+ # ✅ Insert static AccountId (Option 2)
135
  data["AccountId"] = "001dL00001ASyPbQAL"
136
 
137
  record_id, error = create_record(sf, object_name, data)