forked from nicoding/Hackathon
-
Notifications
You must be signed in to change notification settings - Fork 0
/
init.py
49 lines (40 loc) · 1.47 KB
/
init.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
45
46
47
48
49
import os
import redis
import pymysql
import json
redis_host = os.getenv("REDIS_HOST", "localhost")
redis_port = int(os.getenv("REDIS_PORT", "6379"))
r = redis.Redis(host=redis_host, port=redis_port, db=0)
# r.flushall()
conn = pymysql.connect(host=os.getenv("DB_HOST", "localhost"),
port=int(os.getenv("DB_PORT", 3306)),
user=os.getenv("DB_USER", "root"),
passwd=os.getenv("DB_PASS", "toor"),
db=os.getenv("DB_NAME", "eleme"))
redis_connectionpool = redis.Redis(connection_pool =
redis.ConnectionPool(host=os.getenv("REDIS_HOST", "localhost"), port=int(os.getenv("REDIS_PORT", 6379))))
# redis_connectionpool = redis.StrictRedis(host=os.getenv("REDIS_HOST", "localhost"), port=int(os.getenv("REDIS_PORT", 6379)))
cur = conn.cursor()
sql = "SELECT * FROM food"
cur.execute(sql)
foodList=[]
food_id_set=set([])
for r in cur:
d={}
d['id']=r[0]
d['stock']=r[1]
d['price']=r[2]
food_id_set.add(r[0])
foodList.append(d)
# r_food[r[0]]=json.dumps(r[1:])
redis_connectionpool.hmset("food:"+str(r[0]), d)
local_db = {}
local_db["users"] = {}
local_db["access_token"] = {}
local_db["orders"] = {}
local_db["carts"] = {}
cur.execute("SELECT id, name, password FROM user")
for id, name, pw in cur.fetchall():
local_db["users"][name] = {"id": id, "username": name, "password": pw}
foodList=json.dumps(foodList,sort_keys=True)
cur.close()