1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#--depends-on commands
#--depends-on config
import datetime, typing
from src import ModuleManager, utils
DATE_YEAR_FORMAT = "%Y-%m-%d"
DATE_FORMAT = "%d-%b"
EXAMPLE_DATE_YEAR = "1995-09-15"
EXAMPLE_DATE = "01-jan"
def _parse(s):
if s.count("-") == 1:
try:
return False, datetime.datetime.strptime(s, DATE_FORMAT)
except ValueError:
return None
else:
try:
return True, datetime.datetime.strptime(s, DATE_YEAR_FORMAT)
except ValueError:
return None
def _format_year(dt):
return "%s-%s-%s" % (str(dt.year).zfill(4), str(dt.month).zfill(2),
str(dt.day).zfill(2))
def _format_noyear(dt):
return datetime.datetime.strftime(dt, DATE_FORMAT)
def _format(years, dt):
if years:
return _format_year(dt)
else:
return _format_noyear(dt)
def _parse_setting(value):
parsed = _parse(value)
if parsed:
years, parsed = parsed
return _format(years, parsed)
else:
raise utils.settings.SettingParseException(
"Please provide either yyyy-mm-dd or dd-mmm (e.g. %s or %s)" %
(EXAMPLE_DATE_YEAR, EXAMPLE_DATE))
def _apostrophe(nickname):
if nickname[-1].lower() == "s":
return "%s'" % nickname
return "%s's" % nickname
@utils.export("set", utils.FunctionSetting(_parse_setting, "birthday",
"Set your birthday", example=EXAMPLE_DATE_YEAR))
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.birthday")
def birthday(self, event):
"""
:help: Get your, or someone else's, birthday
:usage: [nickname]
:require_setting: birthday
:require_setting_unless: 1
"""
target_user = event["user"]
if event["args"]:
target_user = event["server"].get_user(event["args_split"][0])
birthday = target_user.get_setting("birthday", None)
if not birthday == None:
years, birthday_parsed = _parse(birthday)
birthday_parsed = birthday_parsed.date()
now = datetime.datetime.utcnow().date()
next_birthday = datetime.date(year=now.year,
month=birthday_parsed.month, day=birthday_parsed.day)
if next_birthday < now:
next_birthday = next_birthday.replace(year=next_birthday.year+1)
days = (next_birthday-now).days
days_str = "day" if days == 1 else "days"
age = next_birthday.year-birthday_parsed.year
if days > 0:
if years:
event["stdout"].write("%s is %d in %d %s" % (
target_user.nickname, age, days, days_str))
else:
event["stdout"].write("%s birthday is in %d %s" % (
_apostrophe(target_user.nickname), days, days_str))
else:
if years:
event["stdout"].write("%s is %d today! 🎉" % (
target_user.nickname, age))
else:
event["stdout"].write("%s birthday is today! 🎉" %
_apostrophe(target_user.nickname))
else:
event["stderr"].write("No birthday set for %s" %
target_user.nickname)
@utils.hook("received.command.birthdays")
def birthdays(self, event):
birthday_settings = event["server"].get_all_user_settings("birthday")
birthdays = {}
today = datetime.datetime.utcnow().date()
for nickname, birthday in birthday_settings:
years, birthday_parsed = _parse(birthday)
birthday_parsed = birthday_parsed.date()
if birthday_parsed.replace(year=today.year) == today:
birthdays[nickname] = [years, today.year-birthday_parsed.year]
if birthdays:
birthdays_str = []
for nickname, (years, age) in birthdays.items():
nickname = event["server"].get_user(nickname).nickname
if years:
birthdays_str.append("%s (%d)" % (nickname, age))
else:
birthdays_str.append(nickname)
event["stdout"].write("Birthdays today: %s" %
", ".join(birthdays_str))
else:
event["stdout"].write("There are no birthdays today")
|
