def modifytt(uid, num, day, timestart, field, perem): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') if field == "Предмет": cur.execute( f'SELECT * FROM timetable WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (num, day, timestart, uid)) results = cur.fetchall() if len(results) == 0: return 228 else: cur.execute( f'UPDATE timetable SET lesson = %s WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (perem, num, day, timestart, uid)) elif field == "Время": cur.execute( f'SELECT * FROM timetable WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (num, day, timestart, uid)) results = cur.fetchall() if len(results) == 0: return 228 else: cur.execute( f'UPDATE timetable SET time_start = %s WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (perem, num, day, timestart, uid)) else: connection.close() connection.close()
def showList(u_id): ret = "" connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'SELECT * FROM todotable WHERE u_id = %s ORDER BY priority', (u_id)) results = cur.fetchall() count = 1 if len(results) == 0: return 228 else: for row in results: date = row["date"] name = row["name"] # num = row["num"] priority = row["priority"] if priority == 1: ret += ("№" + str(count) + ' ' + name + " до " + str(date) + " | ВЫСОКИЙ ПРИОРИТЕТ!" + "\n") else: ret += ("№" + str(count) + ' ' + name + " до " + str(date) + "\n") count += 1 return ret connection.close()
def delNumTt(uid, num): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'DELETE from timetable WHERE num = %s AND u_id = %s', (num, uid)) connection.close()
def delElemToDo(num, u_id): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'DELETE from todotable WHERE num = %s AND u_id = %s', (num, u_id)) connection.close()
def find_by_id(id): try: conn = connect() cur = conn.cursor() cur.execute('SELECT * FROM ARTICLES WHERE id = {id}'.format(id=id)) article = cur.fetchall() return article[0] except (Exception, psycopg2.DatabaseError) as error: print(error)
def creatett(u_id, day, lesson, time_start, num): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'INSERT INTO timetable(u_id, day, lesson, time_start, num) VALUES("{u_id}", "{day}", "{lesson}", "{time_start}", "{num}")' ) connection.commit() connection.close()
def update_lemmatized(text, article_id): try: conn = connect() cur = conn.cursor() cur.execute('UPDATE normalized_articles SET lemmatized = %s WHERE article_id = %s', (text, article_id)) conn.commit() return True except (Exception, psycopg2.DatabaseError) as error: print(error) cur.execute("rollback")
def set_analyzed(id): try: conn = connect() cur = conn.cursor() cur.execute('UPDATE ARTICLES SET analyzed = %s WHERE id = %s', (True, id)) conn.commit() return True except (Exception, psycopg2.DatabaseError) as error: cur.execute("rollback")
def find_all(): try: conn = connect() cur = conn.cursor() cur.execute('SELECT * FROM ARTICLES WHERE analyzed = false') articles = cur.fetchall() return articles except (Exception, psycopg2.DatabaseError) as error: print(error)
def createList(u_id, date, name, priority): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'INSERT INTO todotable(u_id, date, name, priority) VALUES("{u_id}", "{date}", "{name}", "{priority}")' ) connection.commit() connection.close()
def create_article_link(link, article_id ): try: conn = connect() cur = conn.cursor() cur.execute('INSERT INTO normalized_articles (link, article_id) VALUES (%s, %s)', (link, article_id)) conn.commit() return True except (Exception, psycopg2.DatabaseError) as error: print("error:", error) cur.execute("rollback")
def count(): try: conn = connect() cur = conn.cursor() cur.execute( 'SELECT COUNT(*) as count FROM ARTICLES WHERE analyzed = false') count = cur.fetchone() return count[0] except (Exception, psycopg2.DatabaseError) as error: print(error)
def delElemOfCalendar(num, uid): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'DELETE from calendartable WHERE num = %s and u_id = %s', (num, uid)) cur.execute( f'UPDATE calendartable SET num = num - 1 WHERE num > %s and u_id = %s', (num, uid)) connection.close()
def checkerDelElemToDo(num, u_id): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'SELECT * from todotable WHERE num = %s AND u_id = %s', (num, u_id)) results = cur.fetchall() if len(results) == 0: return 228 connection.close()
def get_by_limit(limit): try: conn = connect() cur = conn.cursor() cur.execute( 'SELECT * FROM ARTICLES WHERE analyzed = false LIMIT {limit}'. format(limit=limit)) articles = cur.fetchall() return articles except (Exception, psycopg2.DatabaseError) as error: print(error)
def ttShowChecker(word): length = len(word.split()) if length != 2: return 1337 else: num = word.split()[1] connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'SELECT * FROM timetable WHERE num = %s', (num)) results = cur.fetchall() if len(results) == 0: return 228
def showtt(uid, num, day): ret = "" connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'SELECT * FROM timetable WHERE num = %s AND day = %s AND u_id = %s ORDER BY time_start', (num, day, uid)) results = cur.fetchall() print(len(results)) for row in results: lesson = row["lesson"] timestart = row["time_start"] ret += (str(timestart) + " - " + lesson + "\n") return ret
def deltt(uid, num, day, timestart): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'SELECT * FROM timetable WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (num, day, timestart, uid)) results = cur.fetchall() if len(results) == 0: return 228 else: cur.execute( f'DELETE FROM timetable WHERE num = %s AND day = %s AND time_start = %s AND u_id = %s', (num, day, timestart, uid)) connection.close()
def clDelCheck(word): lenght = len(word.split()) if lenght != 1: return 1337 else: num = word.split()[1] checker = num.isdigit() if checker == False: return 228 else: connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'SELECT * FROM timetable WHERE num = %s', (num)) results = cur.fetchall() if len(results) == 0: return 228
def createCalendar(u_id, time, name, date): connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute(f'SELECT * from calendartable') results = cur.fetchall() if (len(results) == 0): num = 0 else: cur.execute(f'SELECT MAX(num) from calendartable') for row in results: num = row["num"] cur.execute( f'INSERT INTO calendartable(u_id, time, name, date, num) VALUES("{u_id}", "{time}", "{name}", "{date}", "{num + 1}")' ) connection.commit() connection.close()
def showCalendar(frDate, toDate, uid): ret = "" connection = connect() with connection: cur = connection.cursor() cur.execute(f'USE heroku_775a68534d5d9c7') cur.execute( f'SELECT * FROM calendartable WHERE date > %s and date < %s and u_id = %s ORDER BY num', (frDate, toDate, uid)) results = cur.fetchall() if len(results) > 0: for row in results: num = row["num"] time = row["time"] name = row["name"] date = row["date"] ret += ("№" + str(num) + ". " + str(date) + " в " + str(time) + " - " + name + "\n") return ret else: return 228
def __init__(self): self.conn = connect("account") print("ToWebSite Initiated.")
def test_db_conn(self): self.assertIsInstance(connection.connect(), pymongo.database.Database)
def __init__(self): self.db = connect()
def before_request(): connection.connect()