forked from hackyourlife/orakel
-
Notifications
You must be signed in to change notification settings - Fork 0
/
actions.py
166 lines (143 loc) · 3.91 KB
/
actions.py
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
# -*- coding: utf-8 -*-
# vim:set ts=8 sts=8 sw=8 tw=80 noet cc=80:
from utils import oneof
from re import match
import datetime
import uuid as guid
import icmplib
_ISO8601_TIME_FORMAT = '%Y-%m-%dT%H:%M:%S'
_DATE_FORMAT = '%d.%m.%Y'
_TIME_FORMAT = '%H:%M:%S'
def uuid():
return str(guid.uuid4())
def isotime(at=None):
"""Stringify time in ISO 8601 format."""
if not at:
at = utcnow()
st = at.strftime(_ISO8601_TIME_FORMAT)
tz = at.tzinfo.tzname(None) if at.tzinfo else 'UTC'
st += ('Z' if tz == 'UTC' else tz)
return st
def date(at=None):
if not at:
at = utcnow()
st = at.strftime(_DATE_FORMAT)
return st
def time(at=None):
if not at:
at = utcnow()
st = at.strftime(_TIME_FORMAT)
return st
def utcnow():
"""Overridable version of utils.utcnow."""
return datetime.datetime.utcnow()
class Actions(object):
"""
if(/^bist du ein bot\??$/i.test(msg))
return 'Nein';
if(/^(wer|was) bist du\??$/i.test(msg))
return 'das Orakel';
if(/^was machst du\??$/i.test(msg))
return 'existieren';
if(/^wer ist dein meister\??$/i.test(msg))
return '/me hat keinen Meister';
if(/^ist das schlimm\??$/i.test(msg))
return oneof(['ja', 'nein', 'vielleicht']);
if(/^stimmt das\??$/i.test(msg) || /^stimmt das[, ]/i.test(msg))
return oneof(['ja', 'nein']);
if(/^ist das falsch\??/i.test(msg))
return oneof(['nein', 'ja']);
if(/^(wie spät(ist es)?|zeit)\??/i.test(msg))
return new Date();
"""
def __init__(self, storage):
self.storage = storage
self.actions = {
"sag": self.sag,
"merke": self.merke,
"sprich": self.sprich,
"ping": self.ping }
self.directactions = {
"stimmt das": self.stimmt_das,
"ist das schlimm": self.ist_das_schlimm,
"ist das falsch": self.ist_das_falsch,
"zeit": self.time,
"utc": isotime,
"guid": uuid,
"uuid": uuid,
"isotime": isotime,
"datum": self.date }
self.regex = {
"^bist du ein bot\\??$": "Nein",
"^(wer|was) bist du\\??$": "das Orakel" }
self.passive_actions = {
"!ping": self.ping }
self.passive_directactions = {
"!utc": isotime,
"!now": self.now,
"!uuid": uuid,
"!date": self.date,
"!time": self.time }
def sag(self, args):
return args
def merke(self, args):
self.storage["remembered"] = args
return False
def sprich(self, args):
return self.storage["remembered"]
def stimmt_das(self):
return oneof(["ja", "nein"])
def ist_das_schlimm(self):
return oneof(["ja", "nein", "vielleicht"])
def ist_das_falsch(self):
return oneof(["ja", "nein"])
def now(self):
return isotime(datetime.datetime.now())
def date(self):
return date(datetime.datetime.now())
def time(self):
return time(datetime.datetime.now())
def ping(self, addr):
if len(addr.strip()) == 0:
return "pong"
addr = addr.split(" ")[0].strip()
try:
result = icmplib.ping(addr)
if not result:
return "Request timed out"
else:
return "%1.2fms" % (result * 1000)
except:
return None
def active(self, msg, nick, send_message):
direct = msg[:-1].strip() if msg[-1] == '?' else msg
if direct in self.directactions:
action = self.directactions[direct]
result = action if type(action) == str else action()
send_message(result)
return True
tokens = msg.split(" ")
action = tokens[0].strip()
args = " ".join(tokens[1:]).strip()
if action in self.actions:
result = self.actions[action](args)
if result:
send_message(result)
return True
return False
def passive(self, msg, nick, send_message):
direct = msg[:-1].strip() if msg[-1] == '?' else msg
if direct in self.passive_directactions:
action = self.passive_directactions[direct]
result = action if type(action) == str else action()
send_message(result)
return True
tokens = msg.split(" ")
action = tokens[0].strip()
args = " ".join(tokens[1:]).strip()
if action in self.passive_actions:
result = self.passive_actions[action](args)
if result:
send_message(result)
return True
return False