-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.py
91 lines (64 loc) · 2.4 KB
/
util.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
# coding: utf-8
import os
import re
from devpower.util import find_classes
from commands import Command
from plugin import PluginCommand
from notifications import Notification
def find_commands(relative_path, home_dir):
commands = {}
def command_found(klass, cwd):
command = klass()
if command.id:
print cwd
commands[command.id] = PluginCommand(command, cwd, home_dir)
else:
raise Exception("Command id is missing")
return True
cwd = os.path.dirname(__file__)
find_classes(Command, relative_path, cwd, command_found)
return commands
class Parsers(dict):
"""
This dictionary is responsable for handling parsers for each command,
and it's needed by the MailExtractor class. It is handled in a
class so that parsers can be loaded at runtime instead of loading
them all at once.
"""
def __init__(self, commands):
dict.__init__(self)
self.__commands = commands
def __getitem__(self, attr):
if attr not in self.__commands:
raise Exception("Command has no parser registered: %s" % (attr))
return self.__commands[attr].parser
class AuthNotification(Notification):
def __init__(self, cmd_id, sender, recipients, email, data):
Notification.__init__(self)
self.title = "@CMD: %s" % (cmd_id)
self.sender = sender
self.recipients = recipients
self.body["text"] = self.__get_body(email, data.raw)
def __get_body(self, headers, body):
# top to bottom
body = "-->\n\n" + body
body = self.__add_property(body, "auth-key")
body = self.__add_property(body, "reply-to", headers["from"])
body = self.__add_property(body, "subject", "Re: " + headers["subject"])
body = "<--\n" + body
return body
def __add_property(self, body, key, value=""):
prop = "%s: %s" % (key, value)
return prop + "\n" + body
class SuccessNotification(Notification):
def __init__(self, sender, recipients, command, sprops):
Notification.__init__(self)
if "subject" in sprops:
self.title = sprops["subject"]
else:
self.title = "Re: " + command.id
if "reply-to" in sprops:
recipients.append(sprops["reply-to"])
self.recipients = recipients
self.sender = sender
self.body = command.output