def check_stop_message_repetition(user_id, stop_message):
session = session_factory()
target_alarms = session.query(DataBaseAlarm).filter_by(user_id=user_id)
for alarm in target_alarms:
if alarm.stop_message == stop_message:
return True
return False
def search_stop_message(user_id, input_message):
print("search")
session = session_factory()
for id, stop_message in session.query(DataBaseAlarm.user_id,
DataBaseAlarm.stop_message):
if id == user_id and stop_message == input_message:
return True
return False
def update_alarm_activation(user_id, stop_message):
session = session_factory()
target_alarm = session.query(DataBaseAlarm).filter_by(
user_id=user_id, stop_message=stop_message).first()
session = session.object_session(target_alarm)
deactive_message = TextMessage("هشدار {} متوقف شد .".format(
target_alarm.name))
if target_alarm.activation_status == "true":
target_alarm.activation_status = False
session.commit()
session.close()
return deactive_message
def update_alarm_time(alarm):
session = session_factory().object_session(alarm)
target_alarm = session.query(DataBaseAlarm).filter_by(id=alarm.id).first()
temp_time = target_alarm.start_time.split(":")
date_time = jdatetime.datetime(int(temp_time[0]), int(temp_time[1]),
int(temp_time[2]), int(temp_time[3]),
int(temp_time[4]))
date_time = date_time + jdatetime.timedelta(
minutes=int(alarm.repeat_period))
target_alarm.start_time = "{}:{}:{}:{}:{}".format(date_time.year,
date_time.month,
date_time.day,
date_time.hour,
date_time.minute)
session.commit()
session.close()
def update_user_excel_file(user_id):
session = session_factory()
debts = session.query(DataBaseDebt).filter_by(user_id=user_id)
workbook = xlsxwriter.Workbook("Excel-Files/{}.xlsx".format(user_id))
worksheet = workbook.add_worksheet()
# Add a bold format to use to highlight cells.
bold = workbook.add_format({'bold': True})
worksheet.write(0, 0, 'ID', bold)
worksheet.write(0, 1, 'Creditor-Name', bold)
worksheet.write(0, 2, 'Amount', bold)
worksheet.write(0, 3, 'Date', bold)
worksheet.write(0, 4, 'Status', bold)
for debt in debts:
worksheet.write(debt.id, 0, str(debt.id))
worksheet.write(debt.id, 1, debt.creditor_name)
worksheet.write(debt.id, 2, debt.amount)
worksheet.write(debt.id, 3, debt.date)
worksheet.write(debt.id, 4, debt.payment_status)
workbook.close()
session.close()
def save_alarm(alarm_from_bot):
session = session_factory()
alarm = DataBaseAlarm(alarm_from_bot)
session.add(alarm)
session.commit()
session.close()
def delete_alarm(alarm):
session = session_factory().object_session(alarm)
session.delete(alarm)
session.commit()
def get_all_debts():
session = session_factory()
debt_query = session.query(DataBaseDebt)
session.close()
return debt_query.all()
def get_all_alarms():
session = session_factory()
alarm_query = session.query(DataBaseAlarm)
session.expire_on_commit = False
session.close()
return alarm_query.all()
def save_photo(photo_messaage):
session = session_factory()
photo = Photo(photo_messaage)
session.add(photo)
session.commit()
session.close()
def save_debt(debt_from_bot):
session = session_factory()
debt = DataBaseDebt(debt_from_bot)
session.add(debt)
session.commit()
session.close()
def get_photo_by_id(id):
session = session_factory()
return session.query(Photo).filter_by(id=id).first()
def get_photo_id(photo_message):
session = session_factory()
target_photo = session.query(Photo).filter_by(
file_id=photo_message.file_id).first()
return target_photo.id
