-
Notifications
You must be signed in to change notification settings - Fork 1
/
__init__.py
101 lines (90 loc) · 4.23 KB
/
__init__.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
import csv
import shutil
from os.path import dirname, join
from mycroft import MycroftSkill, intent_file_handler
inventoryfile = join(dirname(__file__), 'inventory.txt')
editfile = join(dirname(__file__), 'edit.txt')
class Inventory(MycroftSkill):
def __init__(self):
MycroftSkill.__init__(self)
def initialize(self):
self.register_intent_file('inventory start.intent', self.handle_inventorystart)
self.register_intent_file('inventory read.intent', self.handle_inventoryread)
self.register_intent_file('inventory where.intent', self.handle_inventorywhereis)
self.register_intent_file('inventory what.intent', self.handle_inventorywhatis)
self.register_intent_file('inventory remove.intent', self.handle_inventoryremove)
self.register_intent_file('inventory add.intent', self.handle_inventoryadd)
self.register_intent_file('inventory move.intent', self.handle_inventorymove)
@intent_file_handler('inventory start.intent')
def handle_inventorystart(self, message):
self.speak_dialog('inventory start')
@intent_file_handler('inventory read.intent')
def handle_inventoryread(self, message):
self.speak_dialog('inventory read')
with open(inventoryfile) as csv_file:
csv_reader = csv.reader(csv_file, delimiter=',')
for row in csv_reader:
linemessage=f'{row[0]} resides in {row[1]}.'
self.speak(linemessage)
@intent_file_handler('inventory where.intent')
def handle_inventorywhereis(self, message):
self.obj = message.data.get('obj')
with open(inventoryfile) as csv_file:
csv_reader = csv.reader(csv_file, delimiter=',')
for row in csv_reader:
if row[0] == self.obj:
linemessage = self.obj+ ' resides in '+row[1]
self.speak(linemessage)
@intent_file_handler('inventory what.intent')
def handle_inventorywhatis(self, message):
self.loc = message.data.get('loc');
locationlist=[];
s = ', '
with open(inventoryfile) as csv_file:
csv_reader = csv.reader(csv_file, delimiter=',')
for row in csv_reader:
if row[1] == self.loc:
locationlist.append(row[0])
if len(locationlist) > 0:
linemessage = self.loc+ ' contains ' + s.join(locationlist)
self.speak(linemessage)
else:
self.speak('There is nothing there.')
@intent_file_handler('inventory remove.intent')
def handle_inventoryremove(self, message):
self.obj = message.data.get('obj');
with open(inventoryfile, 'r') as inp, open(editfile, 'w') as out:
writer = csv.writer(out)
for row in csv.reader(inp):
if row[0] != self.obj:
writer.writerow(row)
self.speak('Removed')
shutil.copyfile(editfile,inventoryfile)
@intent_file_handler('inventory add.intent')
def handle_inventoryadd(self, message):
self.obj = message.data.get('obj');
self.loc = message.data.get('loc');
addlist=[(self.obj,self.loc)]
with open(inventoryfile, 'r') as inp, open(editfile, 'w') as out:
writer = csv.writer(out)
writer.writerows(addlist)
for row in csv.reader(inp):
writer.writerow(row)
self.speak("Added")
shutil.copyfile(editfile,inventoryfile)
@intent_file_handler('inventory move.intent')
def handle_inventorymove(self, message):
self.obj = message.data.get('obj');
self.loc = message.data.get('loc');
addlist=[(self.obj,self.loc)]
with open(inventoryfile, 'r') as inp, open(editfile, 'w') as out:
writer = csv.writer(out)
for row in csv.reader(inp):
if row[0] == self.obj:
writer.writerows(addlist)
self.speak('Moved')
else:
writer.writerow(row)
shutil.copyfile(editfile,inventoryfile)
def create_skill():
return Inventory()