improved logic in xlsx

This commit is contained in:
Vladyslav Shatrovskyi 2024-02-10 17:25:31 +02:00
parent 0d58cc10e0
commit bd37bc5e67
2 changed files with 3 additions and 4 deletions

View File

@ -52,6 +52,3 @@ class Price(Model):
class Meta:
database = db
# Connect to the database and create tables
db.create_tables([Price])

View File

@ -5,6 +5,8 @@ from models import Price, db
def spreadsheet_transform():
db.create_tables([Price])
try:
# Select the first worksheet
worksheet = spreadsheet.get_worksheet(0)
@ -12,6 +14,7 @@ def spreadsheet_transform():
values = worksheet.get_all_values()
# Convert values to DataFrame
df = pd.DataFrame(values[1:], columns=values[0])
df.replace('\xa0', ' ', regex=True, inplace=True)
except Exception as e:
print(f"Error retrieving data from Google Spreadsheet: {e}")
return
@ -50,7 +53,6 @@ def spreadsheet_transform():
"Цена_опенкарт_крупнопт-ТОВ_os": "Цена_опенкарт_крупнопт_ТОВ_os",
"Скидка дропPRO %": "СкидкаропPRO_процент"
}
# Iterate over rows and save to the database
try:
# Iterate over rows and save to the database