Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,6 @@ from utils import (
|
|
| 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,22 +116,11 @@ def migrate_to_salesforce(pdf_paths, object_name):
|
|
| 117 |
if error:
|
| 118 |
log_failure(pdf_path, object_name, error)
|
| 119 |
return f"Error: {error}"
|
|
|
|
| 120 |
|
| 121 |
-
|
| 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)
|
| 138 |
if error:
|
|
|
|
| 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 |
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 |
+
# β
Required Salesforce IDs added as per Option 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 122 |
data["AccountId"] = "001dL00001ASyPbQAL"
|
| 123 |
+
data["OwnerId"] = "005dL00000f9B0l"
|
| 124 |
|
| 125 |
record_id, error = create_record(sf, object_name, data)
|
| 126 |
if error:
|