From bd37bc5e67d6f17f0bcaecdf681b486cad4598c6 Mon Sep 17 00:00:00 2001
From: Vladyslav Shatrovskyi <shatrovskyi.vlad@gmail.com>
Date: Sat, 10 Feb 2024 17:25:31 +0200
Subject: [PATCH] improved logic in xlsx

---
 models.py                 | 3 ---
 xlsx_to_db_transformer.py | 4 +++-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/models.py b/models.py
index c180857..9d9c8f7 100644
--- a/models.py
+++ b/models.py
@@ -52,6 +52,3 @@ class Price(Model):
 
     class Meta:
         database = db
-
-# Connect to the database and create tables
-db.create_tables([Price])
diff --git a/xlsx_to_db_transformer.py b/xlsx_to_db_transformer.py
index f543e0c..cb4eb8e 100644
--- a/xlsx_to_db_transformer.py
+++ b/xlsx_to_db_transformer.py
@@ -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