DmitrMakeev
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -100,7 +100,7 @@ for db in DATABASES:
|
|
100 |
|
101 |
|
102 |
|
103 |
-
|
104 |
"username": "name",
|
105 |
"phone": "phone",
|
106 |
"email": "email",
|
@@ -122,7 +122,6 @@ template = {
|
|
122 |
|
123 |
|
124 |
|
125 |
-
|
126 |
# Функция для выполнения HTTP запросов
|
127 |
def fetch(url):
|
128 |
try:
|
@@ -781,41 +780,30 @@ def update_or_insert_user(db_name, user_data, mapping_template):
|
|
781 |
cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
|
782 |
user = cursor.fetchone()
|
783 |
|
784 |
-
transformed_data = {}
|
785 |
-
for json_key, db_column in mapping_template.items():
|
786 |
-
value = user_data.get(json_key, "")
|
787 |
-
if isinstance(value, list):
|
788 |
-
transformed_data[db_column] = "-1" if value else "0"
|
789 |
-
else:
|
790 |
-
transformed_data[db_column] = str(value)
|
791 |
-
|
792 |
-
required_fields = [
|
793 |
-
"vk_id", "chat_id", "ws_st", "ws_stop", "web_st", "fin_prog",
|
794 |
-
"b_city", "b_fin", "b_ban", "b_ign", "b_baners", "b_butt", "b_mess",
|
795 |
-
"shop_st", "curator", "pr1", "pr2", "pr3", "pr4", "pr5", "ad_url",
|
796 |
-
"key_pr", "n_con", "canal", "data_t"
|
797 |
-
]
|
798 |
-
|
799 |
-
for field in required_fields:
|
800 |
-
if field not in transformed_data:
|
801 |
-
transformed_data[field] = ""
|
802 |
-
|
803 |
if user:
|
804 |
web_st_value = int(user[0]) + 1
|
805 |
-
|
806 |
update_query = "UPDATE contacts SET "
|
807 |
update_values = []
|
808 |
-
for
|
809 |
-
|
810 |
-
|
|
|
|
|
|
|
811 |
update_query += "web_st = ? WHERE email = ?"
|
812 |
update_values.extend([web_st_value, email])
|
813 |
cursor.execute(update_query, update_values)
|
814 |
else:
|
815 |
-
columns = ', '.join(
|
816 |
-
placeholders = ', '.join('?' for _ in
|
817 |
insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
|
818 |
-
insert_values =
|
|
|
|
|
|
|
|
|
|
|
|
|
819 |
cursor.execute(insert_query, insert_values)
|
820 |
|
821 |
conn.commit()
|
@@ -852,7 +840,7 @@ def send_get_request():
|
|
852 |
# Обновление или добавление каждого пользователя в базу данных data_gc.db
|
853 |
for user_id, user_data in users_meta.items():
|
854 |
user_data['messages'] = messages_json
|
855 |
-
update_or_insert_user(DATABASE_NAME, user_data,
|
856 |
|
857 |
return jsonify({'status': 'User data saved successfully'})
|
858 |
else:
|
|
|
100 |
|
101 |
|
102 |
|
103 |
+
mapping_template = {
|
104 |
"username": "name",
|
105 |
"phone": "phone",
|
106 |
"email": "email",
|
|
|
122 |
|
123 |
|
124 |
|
|
|
125 |
# Функция для выполнения HTTP запросов
|
126 |
def fetch(url):
|
127 |
try:
|
|
|
780 |
cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
|
781 |
user = cursor.fetchone()
|
782 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
783 |
if user:
|
784 |
web_st_value = int(user[0]) + 1
|
|
|
785 |
update_query = "UPDATE contacts SET "
|
786 |
update_values = []
|
787 |
+
for json_key, db_column in mapping_template.items():
|
788 |
+
value = user_data.get(json_key, "")
|
789 |
+
if isinstance(value, list):
|
790 |
+
value = "-1" if value else "0"
|
791 |
+
update_query += f"{db_column} = ?, "
|
792 |
+
update_values.append(str(value))
|
793 |
update_query += "web_st = ? WHERE email = ?"
|
794 |
update_values.extend([web_st_value, email])
|
795 |
cursor.execute(update_query, update_values)
|
796 |
else:
|
797 |
+
columns = ', '.join(mapping_template.values()) + ", web_st"
|
798 |
+
placeholders = ', '.join('?' for _ in mapping_template) + ", ?"
|
799 |
insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
|
800 |
+
insert_values = []
|
801 |
+
for json_key, db_column in mapping_template.items():
|
802 |
+
value = user_data.get(json_key, "")
|
803 |
+
if isinstance(value, list):
|
804 |
+
value = "-1" if value else "0"
|
805 |
+
insert_values.append(str(value))
|
806 |
+
insert_values.append(0)
|
807 |
cursor.execute(insert_query, insert_values)
|
808 |
|
809 |
conn.commit()
|
|
|
840 |
# Обновление или добавление каждого пользователя в базу данных data_gc.db
|
841 |
for user_id, user_data in users_meta.items():
|
842 |
user_data['messages'] = messages_json
|
843 |
+
update_or_insert_user(DATABASE_NAME, user_data, mapping_template)
|
844 |
|
845 |
return jsonify({'status': 'User data saved successfully'})
|
846 |
else:
|