FVDE/ennstatus Changeset - 24980ece978c · Chaos Computer Club Lëtzebuerg Projects

Changeset - 24980ece978c

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Dennis Fink - 11 years ago 2014-01-23 21:55:45
dennis.fink@c3l.lu

simplified json loading functions

1 file changed with 15 insertions and 28 deletions:

ennstatus/status/functions.py

0 comments (0 inline, 0 general)

ennstatus/status/functions.py

➞

Show inline comments

 import os
 import os.path
 import json
 from collections import defaultdict
 from datetime import datetime
 def _load_single_server(name):
     filename = ''.join(['data/', name, '.json'])
     try:
         with open(filename, encoding='utf-8') as f:
             return ''.join(f.readlines())
     except IOError:
         return False
 def _check_server(data):
     date = datetime.strptime(data['last_updated'], '%d-%m-%Y %H:%M:%S')
     now = datetime.utcnow()
     delta = now - date
     if delta.seconds >= 3600:
         status = 'Offline'
     elif delta.seconds >= 1200:
         status = 'Unknown'
     else:
         status = None
     if status is not None:
         for key in ('server_status', 'tor_status'):
             data[key] = status
     filename = os.path.join('data', data['server_name'].lower()+'.json')
     with open(filename, mode='w', encoding='utf-8') as file_object:
         json.dump(data, file_object)
     return data
 def _load_single_server(filename):
     try:
         with open(filename, encoding='utf-8') as f:
             server = json.load(f)
     except IOError:
         return False
     server = _check_server(server)
     return server
 def single_server(name):
     server = _load_single_server(name)
     if server:
         try:
             server = json.loads(server)
             server = _check_server(server)
             return server
         except ValueError:
             return False
     else:
         return False
     filename = ''.join(['data/', name, '.json'])
     return _load_single_server(filename)
 def _get_json_files(root, files):
     for f in files:
         if f.endswith('.json'):
             yield os.path.join(root, f)
 def all_servers():
     for root, _, files in os.walk('data'):
         for f in _get_json_files(root, files):
             with open(f, encoding='utf-8') as fb:
                 try:
                     server = json.load(fb)
                 except ValueError:
                     fb.close()
                     continue
             server = _check_server(server)
             yield server
             yield _load_single_server(f)
 def all_servers_by_type(type):
     for server in all_servers():
         if server['server_type'] == type:
             yield server
 def split_all_servers_to_types():
     servers = defaultdict(list)
     for server in all_servers():
         servers[server['server_type']].append(server)
     return servers

0 comments (0 inline, 0 general)