Flarelyzer/agent.py at master · Javieracost/Flarelyzer · GitHub

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
import csv
import socket
from constants import *
from string import capwords
from subprocess import call
from os import unlink as delete
from traceback import print_exc as stacktrace

interesting = {'rubber cap', 'heat core', 'terra mantle', 'steel boots', 'terra legs', 'dreaded cleaver', "butcher's axe", 'mercenary sword',
               'glooth amulet', 'giant shimmering pearl', 'terra amulet', 'terra hood', 'terra boots', 'glooth cape', 'glooth axe',
               'glooth club', 'glooth blade', 'glooth bag', 'green gem', 'skull staff', 'metal bat', 'gearwheel chain',
               'crown armor', 'royal helmet', 'medusa shield', 'tower shield', 'giant sword', 'sacred tree amulet',
               'zaoan armor', 'zaoan helmet', 'zaoan legs', 'zaoan shoes',
               'deepling squelcher', 'deepling staff', 'necklace of the deep', 'ornate crossbow', 'guardian axe',
               'foxtail', 'heavy trident', "warrior's shield", "warrior's axe",
               'magic plate armor', 'golden legs', 'mastermind shield', 'fire axe', 'demon shield', 'giant sword','demon trophy',
               'demonrage sword', 'gold ring', 'platinum amulet', 'magma legs', 'amber staff', 'onyx flail', 'fire sword', 'magma monocle',
               'magma boots', 'ruthless axe', 'wand of inferno', 'gold ingot',
               'cheese'
               }

notif_time = 2  # in seconds


with open('Database/pluralMap.csv', mode='r') as pluralFile:
    csvFile = csv.reader(pluralFile)
    pluralMap = {row[0]: row[1] for row in csvFile}
    print 'pluralMap loaded'

sockfile = '/tmp/flarelyzer.sock'


def notify(title, msg):
    call(['notify-send', '--urgency=low', '--expire-time=' + str(notif_time * 1000), title, msg])


def quit():
    global sockfile
    try:
        print 'stopping memory scanner'
        client.sendall('QUIT')
        client.recv(10)
    except: pass
    finally:
        print '--Notification agent closed--'
        client.close()
        delete(sockfile)
        notify('Flarelyzer', 'Closed!')
        #print '--Notification agent closed--'
    exit()


def process_loot(loot):
    try:
        loot = map(lambda x: x[1:], loot)
        loot_amounts = dict()
        for i in xrange(len(loot)):
            lootables = loot[i].split()
            if not lootables:
                #print 'skipping strange loot message: ', lootables
                continue
            loot_start = lootables[0]
            if loot_start.isdigit():
                loot[i] = loot[i][loot[i].find(' ') + 1:]
                if loot[i] in pluralMap:
                    loot[i] = pluralMap[loot[i]]
                else:
                    for suffix in plural_suffixes:
                        if loot[i].endswith(suffix):
                            loot[i].replace(suffix, plural_suffixes[suffix])
                            break
                    else:
                        for word in plural_words:
                            if loot[i].startswith(word):
                                loot[i].replace(word, plural_words[word])
                                break
                loot_amounts[loot[i]] = loot_start
            else:
                if loot_start in ['a', 'an']:
                    loot[i] = loot[i][loot[i].find(' ') + 1:]
                loot_amounts[loot[i]] = '0'
        return loot, loot_amounts
    except:
        print 'loot parser error!'
        stacktrace()

interesting = set(map(str.lower, interesting))
print 'Creating temporary socket file...'
agent = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
agent.bind(sockfile)
agent.listen(1)
print 'waiting for client...'
client, addr = agent.accept()
try:
    while True:
        full_msg = client.recv(1024)
        try:
            client.sendall('ACK')
        except IOError:
            quit()
        if full_msg == 'ATTACHED':
            notify('Flarelyzer', 'Started successfully!')
            continue
        elif full_msg == 'NEXT':
            continue
        typeInd = full_msg.find('Loot of ') + 8
        monsterInd = typeInd
        if full_msg.find('a', typeInd) == -1:  # Not a valid loot message
            #print 'skipping invalid loot message: ', full_msg
            continue
        elif full_msg[typeInd] == 'a':  # not an 'a' if its the loot of a boss
            monsterInd = typeInd + 2
        monster = full_msg[monsterInd:full_msg.rfind(':')]
        loot = full_msg[full_msg.rfind(':') + 1:].split(',')
        loot, loot_amounts = process_loot(loot)
        loot = map(str.lower, loot)
        valuables = interesting.intersection(loot)
        if valuables:
            lootmsg = ''
            for v in valuables:
                if loot_amounts[v] != '0':
                    lootmsg += loot_amounts[v] + ' '
                lootmsg += capwords(v) + ', '
            else:
                lootmsg = lootmsg[:-2]
            notify(monster.title(), lootmsg)
except KeyboardInterrupt:
    pass
except Exception, e:
    print 'Notification agent error!'+str(e)
    stacktrace()
    quit()