def get_user_tz(bot, trigger):
"""Gets a user's preferred time zone; will show yours if no user specified."""
nick = trigger.group(2)
if not nick:
nick = trigger.nick
nick = nick.strip()
zone = get_nick_timezone(bot.db, nick)
if zone:
bot.say('%s\'s time zone is %s.' % (nick, zone))
else:
bot.say('%s has not set their time zone' % nick)
def spacex(bot, trigger):
"""Fetches next scheduled SpaceX rocket launch."""
args = trigger.group(2)
if args: args = args.split()
zone = None
if args:
tmp_args = args
for idx, arg in enumerate(tmp_args):
if arg.strip().lower() == "--utc":
zone = "UTC"
args.pop(idx)
channel_or_nick = tools.Identifier(trigger.nick)
zone = zone or get_nick_timezone(bot.db, channel_or_nick)
if not zone:
channel_or_nick = tools.Identifier(trigger.sender)
zone = get_channel_timezone(bot.db, channel_or_nick)
b_url = "https://spacelaunchnow.me/api/3.3.0/launch/upcoming/?format=json&limit=3&search=spacex"
try:
data = requests.get(b_url).json()
except:
return bot.reply("I couldn't fetch data from the API")
if not data.get("results"):
return bot.reply("No results returned from the API")
if args:
tmp_args = " ".join(args)
try:
parsed_data = _parse_results(data,
"SpaceX",
idx=int(tmp_args.strip()) - 1,
tz=zone)
except:
parsed_data = _parse_results(data, "SpaceX", tz=zone)
else:
parsed_data = _parse_results(data, "SpaceX", tz=zone)
for line in parsed_data:
bot.say(line, max_messages=2)
def f_time(bot, trigger):
"""Return the current time.
The command takes an optional parameter: it will try to guess if it's a
nick, a channel, or a timezone (in that order).
If it's a known nick or channel but there is no configured timezone, then
it will complain. If nothing can be found, it'll complain that the argument
is not a valid timezone.
.. seealso::
Function :func:`~sopel.tools.time.format_time` is used to format
the current datetime according to the timezone (if found).
"""
argument = trigger.group(2)
if not argument:
# get default timezone from nick, or sender, or bot, or UTC
zone = get_timezone(bot.db, bot.config, None, trigger.nick,
trigger.sender)
else:
# guess if the argument is a nick, a channel, or a timezone
zone = None
argument = argument.strip()
channel_or_nick = tools.Identifier(argument)
# first, try to get nick or channel's timezone
help_prefix = bot.config.core.help_prefix
if channel_or_nick.is_nick():
zone = get_nick_timezone(bot.db, channel_or_nick)
if zone is None and channel_or_nick in bot.users:
# zone not found for a known nick: error case
set_command = '%ssettz <zone>' % help_prefix
if channel_or_nick != trigger.nick:
bot.say('Could not find a timezone for this nick. '
'%s can set a timezone with `%s`' %
(argument, set_command))
else:
bot.say('Could not find a timezone for you. '
'You can set your timezone with `%s`' %
set_command)
return
else:
zone = get_channel_timezone(bot.db, channel_or_nick)
if zone is None and channel_or_nick in bot.channels:
# zone not found for an existing channel: error case
set_command = '%ssetctz <zone>' % help_prefix
bot.say('Could not find timezone for channel %s. '
'It can be set with `.setctz <zone>`. '
'(requires OP privileges)' % argument)
return
# then, fallback on timezone detection
if zone is None:
# argument not found as nick or channel timezone
try:
zone = validate_timezone(argument)
except ValueError:
bot.say('Could not find timezone "%s".' % argument)
return
time = format_time(bot.db, bot.config, zone, trigger.nick, trigger.sender)
bot.say(time)
def fight(bot, trigger):
"""Fetches UFC/MMA fight event information
By default, with no parameters, will fetch the next or in-progress event.
Use --search text/"text to search" to find a specific event.
Use --prelim to fetch the preliminary card if there is one.
Use --prev to find the previous completed event.
Use --schedule/sched to fetch the next 5 events.
Use --date YYYYMMDD to fetch events on a specific date.
"""
options = parseargs(trigger.group(2))
try:
schedule = requests.get(schedule_url + "&league=ufc" if 'ufc' in
trigger.group(1).lower() else schedule_url)
schedule = schedule.json()
except:
return bot.say("I couldn't fetch the schedule")
pv_fight = options.get('--prev') or False
if any(key.startswith('--prelim') for key in options):
card_type = options.get('--prelim') or options.get('--prelims')
elif any(key.startswith('--fullcard') for key in options):
card_type = 'all'
else:
card_type = 'main'
if card_type == 'early':
card_type = 'prelims2'
elif card_type == True:
card_type = 'prelims1'
search = True if options.get('--search') else False
search_string = options.get('--search')
date_search = options.get('--date', '')
if options.get('--utc'):
zone = "UTC"
else:
zone = options.get('--tz') or None
if zone == True:
zone = "US/Eastern"
channel_or_nick = tools.Identifier(trigger.nick)
zone = zone or get_nick_timezone(bot.db, channel_or_nick)
if not zone:
channel_or_nick = tools.Identifier(trigger.sender)
zone = get_channel_timezone(bot.db, channel_or_nick)
if not zone:
zone = "US/Eastern"
event_blacklist = [
"401210113",
"401223006",
"401210554",
"401223027",
"401223444",
"600002600",
"600004344",
"600004560",
]
now = pendulum.now()
inp_fut = []
previous = []
date_event = []
for date, event in schedule['events'].items():
for ev in event:
if not ev['completed'] and ev['status']['detail'] not in [
'Canceled', 'Postponed'
]:
if 'Bellator 238' in ev['name'] or ev['id'] in event_blacklist:
continue
inp_fut.append(ev)
if ev['completed'] and ev['status']['detail'] not in [
'Canceled', 'Postponed'
]:
previous.append(ev)
if date == date_search:
for ev in event:
date_event.append(ev)
if date_event:
events = {"curr": date_event, "prev": []}
else:
events = {
"curr": inp_fut,
"prev": previous[::-1],
}
if options.get('--sched') or options.get('--schedule'):
schd = []
for sched_event in inp_fut[:5]:
venue = ""
if sched_event.get('venue'):
venue = " - {}".format(sched_event['venue']['fullName'])
if sched_event['venue']['address'].get('state'):
venue += " - {}, {}".format(
sched_event['venue']['address']['city'],
sched_event['venue']['address']['state'],
)
else:
venue += " - {}, {}".format(
sched_event['venue']['address']['city'],
sched_event['venue']['address']['country'],
)
schd.append("{} - {}{}".format(
bold(sched_event['name']),
pendulum.parse(sched_event['date']).in_tz(zone).format(
'MMM Do h:mm A zz'), "{}".format(venue)))
for line in schd:
bot.say(line)
return
if not events['curr']:
if not events['prev']:
return bot.say(
"I couldn't find any past, current, or future events!")
else:
event_id = [events['prev'][0]['id']]
else:
if len(events['curr']) > 1:
event_id = [events['curr'][0]['id']]
date1 = pendulum.parse(events['curr'][0]['date'], strict=False)
for ev in events['curr'][1:]:
tmp_date = pendulum.parse(ev['date'], strict=False)
if date1.is_same_day(tmp_date):
event_id.append(ev['id'])
else:
event_id = [events['curr'][0]['id']]
if pv_fight:
event_id = [events['prev'][0]['id']]
if search:
all_events = previous + inp_fut
for event in all_events:
tmp_name = event['name'].lower().replace(" ", "")
if search_string.lower().replace(" ", "") in tmp_name:
event_id = [event['id']]
break
for id_ in event_id:
try:
data = requests.get(event_url.format(event_id=id_))
data = data.json()
except:
return bot.say("I couldn't fetch current event JSON")
event_name = data['event']['name']
event_time = pendulum.parse(data['event']['date']).in_tz(zone)
event_loc = " - {}".format(data['venue']['displayNameLocation']) if \
data.get('venue') else ""
strings = []
strings.append(
f"\x02{event_name}\x02 - {event_time.format('MMM Do h:mm A zz')}{event_loc}"
)
card_names = {
'main': "-- Main Card --",
'prelims1': "-- Prelims --",
'prelims2': "-- Early Prelims --"
}
if data.get('cards'):
if card_type == 'all':
cards = ['prelims2', 'prelims1', 'main']
else:
cards = [card_type]
for card in cards:
if not data['cards'].get(card):
continue
decisions = []
divisions = []
left_strings = []
right_strings = []
strings.append(card_names.get(card, "-- Unknown --"))
if pv_fight or data['cards'][card]['competitions'][0][
'status']['type']['completed']:
# fight over
fights = data['cards'][card]['competitions'][::-1]
for fight in fights:
left_fighter = {}
right_fighter = {}
flg_code = None
for fighter in fight['competitors']:
if fighter['order'] == 1:
left_fighter['name'] = bold(
fighter['athlete']['displayName']
) if fighter['winner'] else fighter['athlete'][
'displayName']
left_fighter['winner_or_loser'] = color(
"W",
'green') if fighter['winner'] else color(
"L", 'red')
flg_code = None
if "TBA" not in left_fighter['name']:
try:
if len(fighter['athlete']['flag']
['alt']) > 3:
for country in countries:
if fighter['athlete']['flag'][
'alt'] == country[
'name']:
flg_code = country[
'alpha-2']
break
else:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
if not flg_code:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
try:
if flg_code.lower() == "en":
left_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
left_fighter[
'flag'] = "{}".format(
flag.flag(flg_code))
except:
if flg_code.lower() == "en":
left_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
left_fighter['flag'] = fighter[
'athlete']['flag']['alt']
except:
left_fighter['flag'] = ""
else:
left_fighter['flag'] = "ЪЈ│№ИЈ"
left_fighter['record'] = fighter.get(
'displayRecord', "-")
else:
right_fighter['name'] = bold(
fighter['athlete']['displayName']
) if fighter['winner'] else fighter['athlete'][
'displayName']
right_fighter['winner_or_loser'] = color(
"W",
'green') if fighter['winner'] else color(
"L", 'red')
flg_code = None
if "TBA" not in right_fighter['name']:
try:
if len(fighter['athlete']['flag']
['alt']) > 3:
for country in countries:
if fighter['athlete']['flag'][
'alt'] == country[
'name']:
flg_code = country[
'alpha-2']
break
else:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
if not flg_code:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
try:
if flg_code.lower() == "en":
right_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
right_fighter[
'flag'] = "{}".format(
flag.flag(flg_code))
except:
if flg_code.lower() == "en":
right_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
right_fighter['flag'] = fighter[
'athlete']['flag']['alt']
except:
right_fighter['flag'] = ""
else:
right_fighter['flag'] = "ЪЈ│№ИЈ"
right_fighter['record'] = fighter.get(
'displayRecord', "-")
left_string = "[{}] {} {} ({})".format(
left_fighter['winner_or_loser'],
left_fighter['name'],
left_fighter['flag'],
left_fighter['record'],
)
right_string = "[{}] {} {} ({})".format(
right_fighter['winner_or_loser'],
right_fighter['name'],
right_fighter['flag'],
right_fighter['record'],
)
left_strings.append(left_string)
right_strings.append(right_string)
try:
decisions.append(
" | {} - {} ({}) [R{}, {}]".format(
fight['status']['type']['shortDetail'],
fight['status']['result']
['shortDisplayName'], fight['judgesScores']
if fight.get('judgesScores') else
fight['status']['result']['description'],
fight['status']['period'],
fight['status']['displayClock']))
except:
pass
divisions.append(" | {}".format(fight['note'])
if fight.get('note') else "")
else:
# fight not over
fights = data['cards'][card]['competitions'][::-1]
for fight in fights:
left_fighter = {}
right_fighter = {}
flg_code = None
round_status = None
for fighter in fight['competitors']:
if fighter['order'] == 1:
left_fighter['name'] = bold(
fighter['athlete']['displayName']
) if fighter['winner'] else fighter['athlete'][
'displayName']
if fight['status']['type']['completed']:
left_fighter['winner_or_loser'] = color(
"W", 'green'
) if fighter['winner'] else color(
"L", 'red')
else:
left_fighter['winner_or_loser'] = ""
if fight.get('status', {}).get('type', {}).get(
'state', '') != "pre":
round_status = fight.get('status', {}).get(
'type', {}).get('shortDetail', '')
flg_code = None
if "TBA" not in left_fighter['name']:
try:
if len(fighter['athlete']['flag']
['alt']) > 3:
for country in countries:
if fighter['athlete']['flag'][
'alt'] == country[
'name']:
flg_code = country[
'alpha-2']
break
else:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
if not flg_code:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
try:
if flg_code.lower() == "en":
left_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
left_fighter[
'flag'] = "{}".format(
flag.flag(flg_code))
except:
if flg_code.lower() == "en":
left_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
left_fighter['flag'] = fighter[
'athlete']['flag']['alt']
except:
left_fighter['flag'] = ""
else:
left_fighter['flag'] = "ЪЈ│№ИЈ"
left_fighter['record'] = fighter.get(
'displayRecord', "-")
else:
right_fighter['name'] = bold(
fighter['athlete']['displayName']
) if fighter['winner'] else fighter['athlete'][
'displayName']
if fight['status']['type']['completed']:
right_fighter['winner_or_loser'] = color(
"W", 'green'
) if fighter['winner'] else color(
"L", 'red')
else:
right_fighter['winner_or_loser'] = ""
if fight.get('status', {}).get('type', {}).get(
'state', '') != "pre":
round_status = fight.get('status', {}).get(
'type', {}).get('shortDetail', '')
flg_code = None
if "TBA" not in right_fighter['name']:
try:
if len(fighter['athlete']['flag']
['alt']) > 3:
for country in countries:
if fighter['athlete']['flag'][
'alt'] == country[
'name']:
flg_code = country[
'alpha-2']
break
else:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
if not flg_code:
flg_code = fighter['athlete'][
'flag']['alt'][:2]
try:
if flg_code.lower() == "en":
right_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
right_fighter[
'flag'] = "{}".format(
flag.flag(flg_code))
except:
if flg_code.lower() == "en":
right_fighter[
'flag'] = "\x02ЪЈ┤заЂДзаЂбзаЂЦзаЂ«заЂДзаЂ┐\x02"
else:
right_fighter['flag'] = fighter[
'athlete']['flag']['alt']
except:
right_fighter['flag'] = ""
else:
right_fighter['flag'] = "ЪЈ│№ИЈ"
right_fighter['record'] = fighter.get(
'displayRecord', "-")
left_string = "{}{} {} ({})".format(
"{}".format('[' + left_fighter['winner_or_loser'] +
'] ' if left_fighter['winner_or_loser']
else ''),
left_fighter['name'],
left_fighter['flag'],
left_fighter['record'],
)
right_string = "{}{} {} ({})".format(
"{}".format(
'[' + right_fighter['winner_or_loser'] + '] '
if right_fighter['winner_or_loser'] else ''),
right_fighter['name'],
right_fighter['flag'],
right_fighter['record'],
)
left_strings.append(left_string)
right_strings.append(right_string)
try:
decisions.append(
" | {} - {} ({}) [R{}, {}]".format(
fight['status']['type']['shortDetail'],
fight['status']['result']
['shortDisplayName'], fight['judgesScores']
if fight.get('judgesScores') else
fight['status']['result']['description'],
fight['status']['period'],
fight['status']['displayClock']))
except:
if round_status:
decisions.append(" | {} {}".format(
bold(color("LIVE", colors.GREEN)),
round_status))
else:
pass
divisions.append(" | {}".format(fight['note'])
if fight.get('note') else "")
replies = []
l_padding = 0
r_padding = 0
for string in left_strings:
if len(_stripFormatting(string)) >= l_padding:
l_padding = len(string)
for string in right_strings:
if len(_stripFormatting(string)) >= r_padding:
r_padding = len(string)
l_padding += 6
r_padding += 6
for idx, string in enumerate(left_strings):
if "[" not in string:
ppad = 6
else:
ppad = 2
try:
dec = decisions[idx]
except:
dec = ""
strings.append(
"{:{l_padding}} vs {:{r_padding}}{}{}".format(
left_strings[idx],
right_strings[idx],
dec,
divisions[idx] if divisions else "",
l_padding=l_padding if "\x02" in left_strings[idx]
else l_padding - ppad,
r_padding=r_padding if "\x02" in right_strings[idx]
else r_padding - ppad))
for string in strings:
bot.write(['PRIVMSG', trigger.sender], string)
return
