diff --git a/reading.py b/reading.py index df9fd0b..7003014 100644 --- a/reading.py +++ b/reading.py @@ -1,12 +1,30 @@ -import sqlite3, sys +import sqlite3, sys, time from database import * +from datetime import datetime +now = datetime.now() conn = sqlite3.connect('database.db') cursor = conn.cursor() +dt_string = now.strftime("%d/%m/%Y %H:%M:%S") +def first_read_mid(uid, mid): + print("---- FIRST READ OF USER -----") + data = {"uid" : uid, + "mid" : mid, + "date" : dt_string, + "chap" : 0 } + + print("UserID : %s" % data['uid']) + print("MangID : %s" % data['mid']) + print("Date : %s" % data['date']) + print("Chap : %s" % 0) + cursor.execute("""INSERT INTO readings(uid, mid, date, chap) VALUES(:uid, :mid, :date, :chap)""", data) + conn.commit() + return + def get_last_read(uid, mid): cursor.execute("SELECT chap FROM readings WHERE uid = '{}' AND mid = '{}';".format(uid, mid) ) rows = cursor.fetchall() @@ -14,7 +32,6 @@ def get_last_read(uid, mid): return 0 return rows[0][0] - def chapter_read_plus(uid, mid, number): username = get_dbusername(uid) title = get_dbtitle(mid) @@ -23,12 +40,15 @@ def chapter_read_plus(uid, mid, number): print("Manga : %s" % str(mid)) print("Last Read: %s " % get_last_read(uid, mid)) + if get_last_read(uid, mid) == 0 : + print("No value in database, Need to create the column") + first_read_mid(uid, mid) - get_last_release(mid) new_value = int(get_last_read(uid, mid)) + int(number) print("New Value : %s" % str(new_value)) - cursor.execute("UPDATE readings SET chap = '{}' WHERE mid = '{}' AND uid = '{}';".format(new_value, mid, uid) ) + cursor.execute("UPDATE readings SET chap = '{}', date = '{}' WHERE mid = '{}' AND uid = '{}';".format(new_value, dt_string, mid, uid) ) conn.commit() + return def chapter_read_minus(uid, mid, number): username = get_dbusername(uid) @@ -38,12 +58,16 @@ def chapter_read_minus(uid, mid, number): print("Manga : %s" % str(mid)) print("Last Read: %s " % get_last_read(uid, mid) ) - get_last_release(mid) + if get_last_read(uid, mid) == 0 : + print("No value in database, Creating the column...") + first_read_mid(uid, mid) + + return + new_value = int(get_last_read(uid, mid)) - int(number) print("New Value : %s" % str(new_value)) cursor.execute("UPDATE readings SET chap = '{}' WHERE mid = '{}' AND uid = '{}';".format(new_value, mid, uid) ) - conn.commit() def chapter_read_manual(uid, mid, chap): @@ -66,9 +90,9 @@ def chapter_read_manual(uid, mid, chap): def main(): uid = input("UserID : ") mid = input("MangaID : ") - """number = input("Nombre : ") - chapter_read_minus(uid, mid, number)""" - print(get_last_read(uid, mid)) + number = input("Nombre : ") + """chapter_read_minus(uid, mid, number)""" + print(chapter_read_plus(uid, mid, number))