def accept_option(bot: Bot, update: Update):
new_value = update.effective_message.text
config = Config.get()
option = OPTIONS_CACHE[update.effective_user.id]
if option == "max_busy_time":
config.set_max_busy_time(int(new_value))
elif option == "dispatcher_chat_id":
config.set_dispatcher_chat_id(int(new_value))
elif option == "check_drivers_interval":
config.set_check_drivers_interval(int(new_value))
elif option == "lunch_timeout":
config.set_lunch_timeout(int(new_value))
update.effective_chat.send_message(
f"Опции {option} установлено новое значение: {new_value}.")
return ConversationHandler.END
def show_config(bot: Bot, update: Update):
if not Taxopark.is_admin(update.effective_user.id):
return
conf = Config.get()
options = [(key, getattr(conf, key)) for key in conf
if key in Config.translation_map]
keyboard = [[
InlineKeyboardButton(
f"{Config.translation_map[key]}: {val}", callback_data=key)
] for key, val in options]
reply_markup = InlineKeyboardMarkup(keyboard)
update.effective_chat.send_message(
"Выберите опцию:", reply_markup=reply_markup)
return STATE_CHOOSE_OPTION
def process_supervision(bot: Bot, job: Job):
logger.info("checking drivers")
start = monotonic()
drivers = Taxopark.get_all_drivers_from_map()
for driver in drivers:
punisher = Punisher(bot)
punisher.punish_driver(driver)
config = Config.get()
interval = config.check_drivers_interval * 60
end = monotonic()
extra_time = interval - (end - start)
if extra_time > 0:
logger.info("supervisor will sleep for %i seconds", extra_time)
sleep(extra_time)
def punish_driver(self, driver: Driver):
conf = Config.get()
name, surname, status = driver.name, driver.surname, driver.status.value
if not driver.busy:
logger.info('driver %s %s not busy, his status is - "%s"', name,
surname, status)
return
busy_minutes = round(
(datetime.utcnow() - driver.status.set_at).total_seconds() / 60)
if busy_minutes <= conf.max_busy_time:
logger.info(
"skip driver %s %s - he is busy for only %i minutes. "
"it is acceptable",
name,
surname,
busy_minutes,
)
return
if not Taxopark.is_registered(driver):
logger.info("skip unregistered driver %s %s", name, surname)
return
payload = Taxopark.get_payload(driver)
payload.update_timeout()
if payload.timeout or payload.timeout > 1:
logger.info("skip driver %s %s which has timeout %i", name,
surname, payload.timeout)
return
payload.increment_penalties()
punishment = Punishment(payload.penalty)
if punishment.is_warning:
# Taxopark.call_driver(driver.name, punishment.message)
self._send_warning(punishment.message,
driver,
reply_markup=punishment.button)
elif punishment.is_call_dispatcher:
self._call_dispatcher(punishment.message, driver)
Taxopark.add_timeout(driver, int(punishment.update_timeout))
def lunch_request(bot: Bot, update: Update):
driver = Taxopark.get_driver(tg_id=update.effective_user.id)
conf = Config.get()
last_lunch_request_time = driver.last_lunch_request_time
if not last_lunch_request_time or driver.lunch_count < 2:
update.effective_chat.send_message("Обед одобрен.")
Taxopark.add_timeout(driver, conf.lunch_timeout)
driver.lunch_count += 1
driver.last_lunch_request_time = datetime.now()
driver.save()
elif datetime.now().day - driver.last_lunch_request_time.day >= 1:
driver.lunch_count = 0
driver.last_lunch_request_time = datetime.now()
driver.save()
else:
update.effective_chat.send_message(
"Вы уже взяли два обеда. Это максимум в сутки.")
def let_go_for_lunch(bot: Bot, update: Update):
if update.effective_user.id != Taxopark.get_dispatcher_chat_id():
return
query = update.callback_query
data = query.data
answer = {'+': 'Yes', '-': 'No'}[data[0]]
driver_id = data[1:]
driver = Taxopark.get_driver(driver_id=driver_id)
conf = Config.get()
tout = conf.lunch_timeout
if answer == 'Yes':
Taxopark.add_timeout(driver, tout)
bot.send_message(
driver.tg.id,
f"Обед одобрен. Добавлено {tout} минут свободного времени.")
else:
bot.send_message(driver.tg.id, "Ваш запрос на обед отклонен.")
update.callback_query.answer("Ваше решение передано водителю.")
def setup_general_config():
'''
Is used to populate the database if it has been reset.
:return:
'''
long_string = ConfigType.get(type="long_string")
long_string = ConfigType(
type="long_string") if long_string is None else long_string
string = ConfigType.get(type="string")
string = ConfigType(type="string") if string is None else string
number = ConfigType.get(type="number")
number = ConfigType(type="number") if number is None else number
config_list = [{
"key":
"frontend_url",
"value":
"localhost:3001",
"config_type":
string,
"description":
"The url to this frontend page (used for links in the emails)"
}, {
"key":
"archive_base_url",
"value":
"localhost:3001/api/archive/download",
"config_type":
string,
"description":
"The base url to download archives from (used in emails to the board)"
}, {
"key":
"document_template_url",
"value":
"https://www.overleaf.com/read/ddjdhxnkxttj",
"config_type":
string,
"description":
"The template overleaf document for the different reports (used in the emails)"
}, {
"key": "gotify_url",
"value": "http://*****:*****@chalmers.it",
"config_type": string,
"description": "The email to the secretary"
}, {
"key": "board_email",
"value": "*****@*****.**",
"config_type": string,
"description": "The email to the board"
}, {
"key": "group_email_domain",
"value": "@chalmers.it",
"config_type": string,
"description": "The domain to send emails to"
}, {
"key": "from_email_address",
"value": "*****@*****.**",
"config_type": string,
"description": "The email to send from"
}, {
"key":
"mail_to_groups_subject",
"value":
"Dokument till sektionsmöte den {day}/{month}",
"config_type":
string,
"description":
"""
The subject for "regular" email sendout (that goes out to all active groups that have documents to turn in for the meeting). \n
Description of the formatting values: \n
- {group_name} = The display name of the group \n
- {meeting_day} = The day of month for the meeting \n
- {meeting_month} = The month (number) of the meeting \n
- {deadline_time} = The deadline time (hh:mm) \n
- {deadline_date} = The deadline date (dd/mm) \n
- {task_list} = A list of the tasks that the group should upload \n
- {frontend_url} = The url to the website \n
- {group_code} = Their unique code \n
- {template_url} = The document (overleaf) template url \n
- {secretary_email} = The email to the secretary \n
- {board_display_name} = The display name of the board \n
- {board_email} = The email to the board
"""
}, {
"key":
"mail_to_groups_message",
"value":
"\nHej {group_name}!\n\nDen {meeting_day}/{meeting_month} är det dags för sektionsmöte och senast {deadline_time} den {deadline_date} behöver ni lämna in "
"följande dokument: {task_list}\nDetta görs på sidan: {frontend_url}\nAnge koden: {group_code}\n\nMall för vissa "
"dokument finns här: {template_url}\nGör en kopia av projektet (Menu -> Copy Project) och fyll i.\n\nOm ni har "
"några frågor eller stöter på några problem kan kan ni kontakta mig på {secretary_email} eller hela {board_display_name} på {board_email} "
": ).",
"config_type":
long_string,
"description":
"""
The body of the "regular" emails (the ones that are sent to all the active groups that should turn in documents for the meeting). \n
Description of the formatting values: \n
- {group_name} = The display name of the group \n
- {meeting_day} = The day of month for the meeting \n
- {meeting_month} = The month (number) of the meeting \n
- {deadline_time} = The deadline time (hh:mm) \n
- {deadline_date} = The deadline date (dd/mm) \n
- {task_list} = A list of the tasks that the group should upload \n
- {frontend_url} = The url to the website \n
- {group_code} = Their unique code \n
- {template_url} = The document (overleaf) template url \n
- {secretary_email} = The email to the secretary \n
- {board_display_name} = The display name of the board \n
- {board_email} = The email to the board
"""
}, {
"key":
"mail_to_board_subject",
"value":
"Dokument för sektionsmöte {meeting_number} lp {meeting_lp}",
"config_type":
string,
"description":
"""
The subject of the email that is sent to the board upon reaching the deadline. \n
Description of the formatting values: \n
- {board_name} = The display name of the board \n
- {meeting_number} = The number of the meeting (usually 0) \n
- {meeting_lp} = The study period of the meeting \n
- {meeting_archive_url} = A link to the archive download \n
- {secretary_email} = The email to the secretary
"""
}, {
"key":
"mail_to_board_message",
"value":
"\nHej {board_name}!\n\nDeadlinen för dokumentinsamling till sektionsmöte {meeting_number} i lp {meeting_lp} är nu nådd.\nFör "
"nedladdning av dessa dokument klicka på denna länk: {meeting_archive_url}\n\nVid frågor, kontakta sekreteraren på {secretary_email}",
"config_type":
long_string,
"description":
"""
The contents of the email that is sent out to the board upon reaching the deadline. \n
Description of the formatting values: \n
- {board_name} = The display name of the board \n
- {meeting_number} = The number of the meeting (usually 0) \n
- {meeting_lp} = The study period of the meeting \n
- {meeting_archive_url} = A link to the archive download \n
- {secretary_email} = The email to the secretary
"""
}, {
"key":
"mail_for_stories_subject",
"value":
"Dokument för sektionsmöte {meeting_number} lp {meeting_lp}",
"config_type":
string,
"description":
"""
The subject of the email that is sent to the "story groups" (i.e. the groups that needs to turn in eberattelser / vberattelser. \n
Description of the formatting values: \n
- {group_name_year} = Display name of the group. \n
- {meeting_day} = The day of month that the meeting will take place \n
- {meeting_month} = The month (number) of the meeting \n
- {deadline_time} = The deadline time \n
- {deadline_date} = The deadline date \n
- {task_list} = A list of the tasks that the group will have to turn in. \n
- {frontend_url} = A url to the frontend (upload page) \n
- {group_code} = Their unique code \n
- {template_url} = A link the overleaf template for the documents. \n
- {secretary_email} = The email to the secretary \n
- {board_display_name} = The display name of the board \n
- {board_email} = The email to the board \n
- {meeting_number} = The number of the meeting that study period (usually 0) \n
- {meeting_lp} = The study period
"""
}, {
"key":
"mail_for_stories",
"value":
"\nHej {group_name_year}!\n\nDen {meeting_day}/{meeting_month} är det dags för sektionsmöte och senast {deadline_time} den {deadline_date} behöver ni lämna in "
"följande dokument: {task_list}\nDetta görs på sidan: {frontend_url}\nAnge koden: {group_code}\n\nMall för vissa "
"dokument finns här: {template_url}\nGör en kopia av projektet (Menu -> Copy Project) och fyll i.\n "
"Kontakta revisorerna på [email protected] för mer information om vad som behövs göras innan ni "
"kan bli rekomenderade att bli ansvarsbefriade.\n\nOm ni har "
"några frågor eller stöter på några problem kan kan ni kontakta mig på {secretary_email} eller hela {board_display_name} på {board_email} "
": ).",
"config_type":
long_string,
"description":
"""
The body of the email that is sent to the "story groups" (i.e. the groups that needs to turn in eberattelser / vberattelser) \n
Description of the formatting values: \n
- {group_name_year} = Display name of the group. \n
- {meeting_day} = The day of month that the meeting will take place \n
- {meeting_month} = The month (number) of the meeting \n
- {deadline_time} = The deadline time \n
- {deadline_date} = The deadline date \n
- {task_list} = A list of the tasks that the group will have to turn in. \n
- {frontend_url} = A url to the frontend (upload page) \n
- {group_code} = Their unique code \n
- {template_url} = A link the overleaf template for the documents. \n
- {secretary_email} = The email to the secretary \n
- {board_display_name} = The display name of the board \n
- {board_email} = The email to the board \n
- {meeting_number} = The number of the meeting that study period (usually 0) \n
- {meeting_lp} = The study period
"""
}, {
"key": "board_display_name",
"value": "styrIT",
"config_type": string,
"description": "The display name of the board"
}, {
"key":
"minutes_after_deadline_to_mail",
"value":
"5",
"config_type":
number,
"description":
"The amount of minutes to wait extra after the deadline before sending the email to the board"
}, {
"key":
"check_for_deadline_frequency",
"value":
"5",
"config_type":
number,
"description":
"The frequence (in minutes) to check if any deadlines have been reached"
}, {
"key":
"possible_years_back_for_stories",
"value":
"5",
"config_type":
number,
"description":
"The number of years back that one should be able to select story groups for (usually 5 due to tax reasons)"
}]
for config in config_list:
conf = Config.get(key=config["key"])
if conf is None:
Config(key=config["key"],
value=config["value"],
config_type=config["config_type"],
description=config["description"])
else:
# Since the only way to change the description is here,
# we always want the db version to be up to date with this list on application restart.
conf.description = config["description"]
# Setup groups and tasks
groups = [{
"codeName": "armit",
"displayName": "ArmIT"
}, {
"codeName": "digit",
"displayName": "digIT"
}, {
"codeName": "fanbarerit",
"displayName": "FanbärerIT"
}, {
"codeName": "fritid",
"displayName": "frITid"
}, {
"codeName": "mrcit",
"displayName": "MRCIT"
}, {
"codeName": "nollkit",
"displayName": "NollKIT"
}, {
"codeName": "prit",
"displayName": "P.R.I.T."
}, {
"codeName": "sexit",
"displayName": "sexIT"
}, {
"codeName": "snit",
"displayName": "snIT"
}, {
"codeName": "styrit",
"displayName": "styrIT"
}, {
"codeName": "equalit",
"displayName": "EqualIT"
}, {
"codeName": "flashit",
"displayName": "FlashIT"
}]
tasks = [{
"codeName": "vplan",
"displayName": "Verksamhetsplan / Operational plan"
}, {
"codeName": "budget",
"displayName": "Budget"
}, {
"codeName": "vrapport",
"displayName": "Verksamhetsrapport / Operational report"
}, {
"codeName": "vberattelse",
"displayName": "Verksamhetsberättelse / Operational story"
}, {
"codeName": "eberattelse",
"displayName": "Ekonomisk Berättelse / Economic story"
}]
for group in groups:
if Group.get(name=group["codeName"]) is None:
new_group = Group(name=group["codeName"],
display_name=group["displayName"])
if GroupYear.get(group=new_group, year="active") is None:
GroupYear(group=new_group, year="active", finished=False)
for task in tasks:
if Task.get(name=task["codeName"]) is None:
Task(name=task["codeName"], display_name=task["displayName"])
commit()
print("Finished loading database data from general config file.")
def get_config_value(config_key: str) -> str:
config = Config.get(key=config_key)
if config is None:
raise Exception("No config found with key {0}".format(config_key))
return config.value
def get_config(key: str) -> Optional[Config]:
return Config.get(key=key)
def get_dispatcher_chat_id(cls) -> int:
conf = Config.get()
return conf.dispatcher_chat_id
