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()
