class HandleCommandArgs:
def __init__(self, bot, update, args):
self.bot = bot
self.update = update
self.args = args
self.bot_constants = BotConstants()
self.bot_variables = BotVariables()
self.strava_telegram_webhooks_resource = StravaTelegramWebhooksResource(
)
def default(self):
pass
def token_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
athlete_details = self.strava_telegram_webhooks_resource.get_athlete(
athlete_id)
if athlete_details:
athlete_token = athlete_details['athlete_token']
message = "Token for {athlete_id}: `{athlete_token}`".format(
athlete_id=athlete_id, athlete_token=athlete_token)
else:
message = "Athlete ID {athlete_id} not found.".format(
athlete_id=athlete_id)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning("More than 1 arguments passed for /token. Args %s",
self.args)
def activate_athlete_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
if self.strava_telegram_webhooks_resource.activate_flag_athlete(
athlete_id=athlete_id):
message = "Successfully activated {athlete_id}.".format(
athlete_id=athlete_id)
else:
message = "Failed to activate {athlete_id}.".format(
athlete_id=athlete_id)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning(
"More than 1 arguments passed for /activate. Args %s",
self.args)
def deactivate_athlete_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
if self.strava_telegram_webhooks_resource.deactivate_flag_athlete(
athlete_id=athlete_id):
message = "Successfully deactivated {athlete_id}.".format(
athlete_id=athlete_id)
else:
message = "Failed to deactivate {athlete_id}.".format(
athlete_id=athlete_id)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning(
"More than 1 arguments passed for /deactivate. Args %s",
self.args)
def update_stats_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
if self.strava_telegram_webhooks_resource.update_stats(athlete_id):
message = "Updating stats for {}..".format(athlete_id)
else:
message = "Failed to trigger update stats for {}".format(
athlete_id)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning(
"More than 1 arguments passed for /deactivate. Args %s",
self.args)
def challenges_refresh_stats_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
message = self.bot_constants.MESSAGE_UPDATE_STATS_CHALLENGES_FAILED
if self.strava_telegram_webhooks_resource.update_challenges_stats(
athlete_id):
message = self.bot_constants.MESSAGE_UPDATE_STATS_CHALLENGES_SUCCESS
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning(
"More than 1 arguments passed for /deactivate. Args %s",
self.args)
def challenges_deauth_command(self):
if len(self.args) == 1:
athlete_id = self.args[0]
if self.strava_telegram_webhooks_resource.challenges_deauth_athlete(
athlete_id):
message = "Successfully deauthorised {athlete_id} from challenges".format(
athlete_id=athlete_id)
else:
message = "Failed to deauthorise {athlete_id} from challenges".format(
athlete_id=athlete_id)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
else:
logging.warning(
"More than 1 arguments passed for /deactivate. Args %s",
self.args)
def process(self):
command = self.update.message.text.split(' ', 1)[0]
options = defaultdict(
lambda: self.default, {
'/token': self.token_command,
'/activate': self.activate_athlete_command,
'/deactivate': self.deactivate_athlete_command,
'/update': self.update_stats_command,
'/challenges_refresh_stats':
self.challenges_refresh_stats_command,
'/challenges_deauth': self.challenges_deauth_command
})
options[command]()
class HandleCommands:
def __init__(self, bot, update, user_data):
self.bot = bot
self.update = update
self.user_data = user_data
self.bot_variables = BotVariables()
self.bot_constants = BotConstants()
self.telegram_user_first_name = self.update.message.from_user.first_name
self.strava_telegram_webhooks_resource = StravaTelegramWebhooksResource(
)
self.aes_cipher = AESCipher(self.bot_variables.crypt_key_length,
self.bot_variables.crypt_key)
self.telegram_username = self.update.message.from_user.username
self.chat_id = self.update.message.chat_id
self.athlete_details = None
self.registration = HandleRegistration(self.bot, self.update,
self.user_data)
def update_user_chat_id(self):
if not self.athlete_details['chat_id'] or int(
self.athlete_details['chat_id']) != int(self.chat_id):
self.strava_telegram_webhooks_resource.update_chat_id(
chat_id=self.chat_id,
athlete_id=self.athlete_details['athlete_id'])
def stats_command(self):
self.user_data.clear()
self.update_user_chat_id()
stats = ProcessStats(self.update)
stats.process()
def refresh_command(self):
self.user_data.clear()
self.update_user_chat_id()
message = self.bot_constants.MESSAGE_UPDATE_STATS_FAILED.format(
first_name=self.telegram_user_first_name)
if self.strava_telegram_webhooks_resource.update_stats(
self.athlete_details['athlete_id']):
message = self.bot_constants.MESSAGE_UPDATE_STATS_STARTED.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
def refresh_all_stats_command(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_UPDATE_STATS_FAILED.format(
first_name=self.telegram_user_first_name)
if self.strava_telegram_webhooks_resource.update_all_stats():
message = self.bot_constants.MESSAGE_UPDATE_STATS_STARTED_ALL.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
def challenges_refresh_all_stats_command(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_UPDATE_STATS_CHALLENGES_FAILED.format(
first_name=self.telegram_user_first_name)
if self.strava_telegram_webhooks_resource.update_challenges_all_stats(
):
message = self.bot_constants.MESSAGE_UPDATE_STATS_CHALLENGES_STARTED.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
def all_athletes_command(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_FETCHING_REGISTERED_ATHLETES.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
all_athletes = self.strava_telegram_webhooks_resource.database_read_all(
self.bot_constants.QUERY_GET_ATHLETES)
sl_no = 0
messages = list()
names = "*List of registered athletes:*\n\n"
for athlete in all_athletes:
sl_no += 1
names += "{sl_no}. [{name}](https://www.strava.com/athletes/{athlete_id})\n".format(
sl_no=sl_no, name=athlete[0], athlete_id=athlete[1])
if sl_no % 25 == 0:
messages.append(names)
names = "*List of registered athletes:*\n\n"
messages.append(names)
for name in messages:
if name != "*List of registered athletes:*\n\n":
self.update.message.reply_text(name,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(name)
def challenges_even_athletes(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_FETCHING_REGISTERED_ATHLETES_EVEN_CHALLENGES.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
results = self.strava_telegram_webhooks_resource.get_even_challenges_athletes(
)
for result in results:
self.update.message.reply_text(result,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(result)
def challenges_odd_athletes(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_FETCHING_REGISTERED_ATHLETES_ODD_CHALLENGES.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
results = self.strava_telegram_webhooks_resource.get_odd_challenges_athletes(
)
for result in results:
self.update.message.reply_text(result,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(result)
def activity_summary_command(self):
self.user_data.clear()
self.user_data['ride_summary'] = {
'athlete_id': self.athlete_details['athlete_id']
}
if self.athlete_details['enable_activity_summary']:
message = self.bot_constants.MESSAGE_ACTIVITY_SUMMARY_SHOULD_DISABLE.format(
first_name=self.telegram_user_first_name)
reply_markup = self.bot_constants.KEYBOARD_ACTIVITY_SUMMARY_DISABLE_CONFIRMATION
else:
message = self.bot_constants.MESSAGE_ACTIVITY_SUMMARY_CONFIRMATION.format(
first_name=self.telegram_user_first_name)
reply_markup = self.bot_constants.KEYBOARD_ENABLE_ACTIVITY_SUMMARY_CONFIRMATION
self.update.message.reply_text(message, reply_markup=reply_markup)
self.strava_telegram_webhooks_resource.send_message(message)
def auto_update_indoor_ride_command(self):
self.user_data.clear()
self.user_data['auto_update_indoor_ride'] = {
'athlete_id': self.athlete_details['athlete_id'],
'athlete_token': self.athlete_details['athlete_token']
}
if self.athlete_details['update_indoor_ride']:
configured_data = ""
if self.athlete_details['update_indoor_ride_data']['name']:
configured_data += "\nActivity Name: {activity_name}".format(
activity_name=self.
athlete_details['update_indoor_ride_data']['name'])
if self.athlete_details['update_indoor_ride_data']['gear_id']:
bike_name = self.strava_telegram_webhooks_resource.get_gear_name(
self.athlete_details['athlete_token'],
self.athlete_details['update_indoor_ride_data']['gear_id'])
configured_data += "\nBike: {bike_name}".format(
bike_name=bike_name)
message = self.bot_constants.MESSAGE_SHOULD_UPDATE_INDOOR_RIDE_DISABLE.format(
first_name=self.telegram_user_first_name,
configuration=configured_data)
reply_markup = self.bot_constants.KEYBOARD_AUTO_UPDATE_INDOOR_RIDE_DISABLE_PROMPT
else:
message = self.bot_constants.MESSAGE_UPDATE_INDOOR_RIDE_CHOOSE_ACTIVITY_NAME.format(
first_name=self.telegram_user_first_name)
reply_markup = self.bot_constants.KEYBOARD_AUTO_UPDATE_INDOOR_RIDE_NAME
self.update.message.reply_text(message, reply_markup=reply_markup)
self.strava_telegram_webhooks_resource.send_message(message)
def challenges_hits_reset(self):
self.user_data.clear()
result = self.strava_telegram_webhooks_resource.challenges_hits_reset()
if result:
message = self.bot_constants.MESSAGE_CHALLENGES_HITS_RESET_SUCCESS.format(
first_name=self.telegram_user_first_name)
else:
message = self.bot_constants.MESSAGE_CHALLENGES_HITS_RESET_FAIL.format(
first_name=self.telegram_user_first_name)
self.update.message.reply_text(message,
parse_mode="Markdown",
disable_web_page_preview=True)
self.strava_telegram_webhooks_resource.send_message(message)
def help_command(self):
self.user_data.clear()
self.update_user_chat_id()
message = self.bot_constants.MESSAGE_HELP_COMMANDS
self.update.message.reply_text(message)
self.strava_telegram_webhooks_resource.send_message(
"Sent help commands.")
def cancel_command(self):
self.user_data.clear()
message = self.bot_constants.MESSAGE_CANCEL_CURRENT_OPERATION
self.update.message.reply_text(message)
self.strava_telegram_webhooks_resource.send_message(message)
def process(self):
self.bot.send_chat_action(chat_id=self.chat_id,
action=telegram.ChatAction.TYPING)
self.athlete_details = self.strava_telegram_webhooks_resource.get_athlete_by_telegram_username(
self.telegram_username)
if self.athlete_details:
command = self.update.message.text
options = defaultdict(
lambda: self.help_command, {
'/start': self.help_command,
'/next': self.help_command,
'/stats': self.stats_command,
'/refresh_stats': self.refresh_command,
'/auto_update_indoor_ride':
self.auto_update_indoor_ride_command,
'/cancel': self.cancel_command,
'/refresh_all_stats': self.refresh_all_stats_command,
'/all_athletes': self.all_athletes_command,
'/challenges_even_athletes': self.challenges_even_athletes,
'/challenges_odd_athletes': self.challenges_odd_athletes,
'/challenges_refresh_all_stats':
self.challenges_refresh_all_stats_command,
'/challenges_hits_reset': self.challenges_hits_reset,
'/activity_summary': self.activity_summary_command,
'/help': self.help_command
})
options[command]()
else:
self.registration.main()