Garmelon/euphoria-plusone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update db structure

Browse source
This commit is contained in:
Joscha 2017-09-08 09:35:48 +00:00
parent 46f85ba04c
commit f24c37a266
1 changed files with 8 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

19
PlusOne.py
View file

@ -8,10 +8,13 @@ import sys
class PointDB(yaboli.Database):
@yaboli.Database.operation
def initialize(conn):
print("INITIALIZING")
cur = conn.cursor()
cur.execute(("CREATE TABLE IF NOT EXISTS Points "
"(nick TEXT, points INTEGER)"))
cur.execute((
"CREATE TABLE IF NOT EXISTS Points ("
"nick TEXT UNIQUE NOT NULL,"
"points INTEGER"
")"
))
conn.commit()
@yaboli.Database.operation
@ -19,14 +22,8 @@ class PointDB(yaboli.Database):
nick = mention_reduced(nick)
cur = conn.cursor()
cur.execute("SELECT * FROM Points WHERE nick=?", (nick,))
if cur.fetchone():
print(f"@{nick}: already in db, updating...")
cur.execute("UPDATE Points SET points=points+1 WHERE nick=?", (nick,))
else:
print(f"@{nick}: not in db, adding...")
cur.execute("INSERT INTO Points (nick, points) VALUES (?, 1)", (nick,))
cur.execute("INSERT OR IGNORE INTO Points (nick, points) VALUES (?, 0)", (nick,))
cur.execute("UPDATE Points SET points=points+1 WHERE nick=?", (nick,))
conn.commit()
@yaboli.Database.operation