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