Oups, add a lot of missing files
This commit is contained in:
parent
d234d83df8
commit
061e73722f
21 changed files with 937 additions and 2 deletions
57
modules/whereis/UpdatedStorage.py
Normal file
57
modules/whereis/UpdatedStorage.py
Normal file
|
|
@ -0,0 +1,57 @@
|
|||
# coding=utf-8
|
||||
|
||||
import socket
|
||||
from datetime import datetime
|
||||
from datetime import timedelta
|
||||
|
||||
from .User import User
|
||||
|
||||
class UpdatedStorage:
|
||||
def __init__(self, url, port):
|
||||
sock = connect_to_ns(url, port)
|
||||
self.users = dict()
|
||||
if sock != None:
|
||||
users = list_users(sock)
|
||||
if users is not None:
|
||||
for l in users:
|
||||
u = User(l)
|
||||
if u.login not in self.users:
|
||||
self.users[u.login] = list()
|
||||
self.users[u.login].append(u)
|
||||
self.lastUpdate = datetime.now ()
|
||||
else:
|
||||
self.users = None
|
||||
sock.close()
|
||||
else:
|
||||
self.users = None
|
||||
|
||||
def update(self):
|
||||
if datetime.now () - self.lastUpdate < timedelta(minutes=10):
|
||||
return self
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
def connect_to_ns(server, port):
|
||||
try:
|
||||
s = socket.socket()
|
||||
s.settimeout(3)
|
||||
s.connect((server, port))
|
||||
except socket.error:
|
||||
return None
|
||||
s.recv(8192)
|
||||
return s
|
||||
|
||||
|
||||
def list_users(sock):
|
||||
try:
|
||||
sock.send('list_users\n'.encode())
|
||||
buf = ''
|
||||
while True:
|
||||
tmp = sock.recv(8192).decode()
|
||||
buf += tmp
|
||||
if '\nrep 002' in tmp or tmp == '':
|
||||
break
|
||||
return buf.split('\n')[:-2]
|
||||
except socket.error:
|
||||
return None
|
||||
Loading…
Add table
Add a link
Reference in a new issue