This repository has been archived by the owner on Oct 29, 2019. It is now read-only.
/
main.py
44 lines (40 loc) · 1.58 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import pymysql.cursors
from raven import Client
import config
import parser
if not config.DEBUG:
client = Client(
dsn=config.sentry_url,
processors=(
'raven.processors.SanitizePasswordsProcessor',
)
)
# Connect to the database
connection = pymysql.connect(host=config.db["host"],
user=config.db["user"],
password=config.db["password"],
db=config.db["db"],
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor,
unix_socket='/var/run/mysqld/mysqld.sock')
try:
with connection.cursor() as cursor:
venue_sql = "SELECT id FROM venues WHERE name = %s"
check_meal_sql = "SELECT id FROM meals WHERE name=%s"
insert_meal_sql = "INSERT INTO meals (name) VALUE (%s)"
insert_sql = 'REPLACE INTO menus (venue, date, meal) VALUES (%s, %s, %s)'
for p in [parser.zuppa, parser.aai]:
cursor.execute(venue_sql, p.name)
venue_id = cursor.fetchone()["id"]
for i in p.get_menus():
cursor.execute(check_meal_sql, (i["name"]))
check = cursor.fetchone()
if check:
meal_id = check["id"]
else:
cursor.execute(insert_meal_sql, (i["name"]))
meal_id = cursor.lastrowid
cursor.execute(insert_sql, (venue_id, i["date"], meal_id))
connection.commit() # save changes
finally:
connection.close()