/
smoker.py
49 lines (39 loc) · 1.35 KB
/
smoker.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
import socket
import time
from storage import packet_size, store, time_sleep, time_smoke
from utils import _print
def process(code, request):
message = ''
while True:
if message != 'ack':
_print('Esperando {}!'.format(store.get(code)['required']))
request.send('need'.encode('UTF-8'))
message = request.recv(packet_size).decode('UTF-8')
if message == 'enable':
_print('Servido!')
time.sleep(time_sleep)
request.send('ack'.encode('UTF-8'))
_print('Armando cigarro!')
time.sleep(time_sleep)
_print('Fumando!!!')
time.sleep(time_smoke)
request.send('enable'.encode('UTF-8'))
elif message == 'ack':
pass
time.sleep(time_sleep)
def init(ip, port, code):
try:
request = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
request.connect((ip, port))
request.send('{}'.format(code).encode('UTF-8'))
time.sleep(time_sleep)
resp = request.recv(packet_size).decode('UTF-8')
if resp == 'accepte':
process(code, request)
else:
_print('Rechazado por el agente.')
request.close()
except KeyboardInterrupt:
_print('Cerrando conexiones...')
request.send('exit'.encode('UTF-8'))
request.close()