members.py

   1 #!/usr/bin/env python3
   2 # -*- coding: utf-8 -*-
   3
   4 import os
   5 import time
   6 import re
   7 import logging
   8 from datetime import datetime
   9 import calendar
  10 import urllib.request
  11 from http.cookiejar import CookieJar
  12
  13 from html_parser import *
  14 import htmlentities
  15
  16 #SERVER
  17 #LOGIN
  18 #PASSWORD
  19 #BASE_URL
  20 HOME = os.environ['HOME']
  21 ARCHIVE = HOME + '/fourmizzz/archive.py'
  22 RESULT = HOME + '/fourmizzz/results.log'
  23
  24 def read_config():
  25     '''
  26     That function will read config.py in fourmizzz directory
  27     and set up globals SERVER, LOGIN, PASSWORD, and BASE_URL.
  28     '''
  29     global SERVER, LOGIN, PASSWORD
  30     global BASE_URL
  31     CONFIG_TEMPLATE='''# Veuillez modifier les lignes ci dessous avec vos paramètres:
  32 SERVER = 's1.fourmizzz.fr'
  33 LOGIN = 'MonIdentifiant'
  34 PASSWORD = 'MonMotDePasse'
  35     '''
  36
  37     sys.path.append(HOME+'/fourmizzz')
  38     try:
  39         import config
  40     except ImportError:
  41         CONFIG = HOME+'/fourmizzz/config.py'
  42         logging.fatal("No configuration file. Creating %s", CONFIG)
  43         f = open(CONFIG, mode='w+', encoding='utf-8')
  44         f.write(CONFIG_TEMPLATE)
  45         f.close()
  46         logging.fatal("Please update this file with your details.")
  47         sys.exit(1)
  48     SERVER = config.SERVER
  49     LOGIN = config.LOGIN
  50     PASSWORD = config.PASSWORD
  51     BASE_URL = 'http://%s' % SERVER
  52
  53 def hide_password_from_log(txt):
  54     return re.sub('mot_passe=.*', 'mot_passe=********', txt)
  55
  56 __opener__ = None
  57 def httpopen(url, post_data=None):
  58     if post_data:
  59         log_post_data = hide_password_from_log(post_data)
  60         post_data = post_data.encode('utf-8') # str->bytes
  61         logging.debug('HTTP POST %s %s', url, log_post_data)
  62     else:
  63         logging.debug('HTTP GET %s', url)
  64     global __opener__
  65     if __opener__ is None:
  66         cookiejar = CookieJar()
  67         __opener__ = urllib.request.build_opener()
  68         __opener__.add_handler(urllib.request.HTTPCookieProcessor(cookiejar))
  69     http_response = __opener__.open(url, post_data)
  70     return http_response
  71
  72 def sleep(seconds):
  73     logging.debug('Waiting %s seconds', seconds)
  74     time.sleep(seconds)
  75
  76 def timestamp():
  77     return calendar.timegm(datetime.now().timetuple())
  78
  79 def number_format(i):
  80     result = ''
  81     while i:
  82         if result:
  83             result = ' ' + result
  84         d3 = i % 1000
  85         i = i//1000
  86         if i:
  87             result = ('%03d' % d3) + result
  88         else:
  89             result = ('%d' % d3) + result
  90     return result
  91
  92
  93 def tdc_get(alliance_tag=None):
  94     #httpresponse = httpopen(BASE_URL + '/')
  95     #html = httpresponse.read().decode('utf-8')
  96     #print(html)
  97
  98     httpresponse = httpopen(BASE_URL + '/index.php?connexion=1', 'serveur=%s&pseudo=%s&mot_passe=%s' % (SERVER, LOGIN, PASSWORD))
  99     #httpresponse = httpopen(BASE_URL + '/index.php?connexion=1', 'serveur=%s&pseudo=%s&mot_passe=%s&connexion=Connexion' % (SERVER, LOGIN, PASSWORD))
 100     html = httpresponse.read().decode('utf-8')
 101     if "redirectReine" not in html:
 102         logging.fatal("Impossible de s'authentifier. Vérifiez vos paramètres dans config.py")
 103         sys.exit(1)
 104     #print('-'*80)
 105     #print(html)
 106
 107     #httpresponse = httpopen(BASE_URL + '/alliance.php?Membres')
 108     #html = httpresponse.read().decode('utf-8')
 109     #print('-'*80)
 110     #print(html)
 111
 112     if not alliance_tag:
 113         httpresponse = httpopen(BASE_URL + '/alliance.php?Membres',
 114             'xajax=membre&xajaxr='+str(timestamp()))
 115         html = httpresponse.read().decode('utf-8')
 116         #print('-'*80)
 117         #print(html)
 118
 119         root = html_parse(html)
 120         table = get_elem(root, 'table')[0]
 121         td = get_elem(table, 'td')[3]
 122         table = get_elem(td, 'table')[0]
 123         #print('-'*80)
 124         #print_idented_tree(table)
 125     else:
 126         httpresponse = httpopen(BASE_URL + '/classementAlliance.php?alliance=%s' % alliance_tag)
 127         html = httpresponse.read().decode('utf-8')
 128
 129         root = html_parse(html)
 130         table = get_elem(root, 'table')[2]
 131         #print('-'*80)
 132         #print_idented_tree(table)
 133
 134     summary = {}
 135     for tr in get_elem(table, 'tr'):
 136         row = get_elem(tr, 'td')
 137         if not alliance_tag and len(row) != 10:
 138             continue
 139         if alliance_tag and len(row) != 6:
 140             continue
 141         #print('-'*10)
 142         #print_idented_tree(tr)
 143         if not alliance_tag:
 144             nick = get_merged_leaf_content(row[3])
 145             tdc = get_merged_leaf_content(row[4]).replace(' ', '')
 146         else:
 147             nick = get_merged_leaf_content(row[2])
 148             tdc = get_merged_leaf_content(row[3]).replace(' ', '')
 149
 150         tdc = int(tdc)
 151         #print(nick, tdc)
 152         summary[nick] = tdc
 153     logging.debug('%s members - total tdc = %s cm²', len(summary), number_format(sum(summary.values())))
 154     return summary
 155
 156
 157 def tdc_compare(oldtdc, newtdc):
 158     changes = {}
 159     txtchanges = []
 160     for nick in newtdc.keys():
 161         told = oldtdc.get(nick, 0)
 162         tnew = newtdc[nick]
 163         if told == tnew:
 164             continue
 165         changes[nick] = { 'old': told, 'new': tnew, 'delta': tnew-told }
 166
 167     for nick in oldtdc.keys():
 168         if nick not in newtdc:
 169             told = oldtdc[nick]
 170             tnew = 0
 171             changes[nick] = { 'old': told, 'new': 0, 'delta': -told }
 172     #for nick, change in changes.items():
 173     #    print(nick, ' - ', change['old'], ' - ', change['new'], ' - ', change['delta'])
 174     #print
 175
 176     if not len(changes):
 177         logging.info('No changes')
 178         return
 179
 180     for nick in changes.keys():
 181         delta = changes[nick]['delta']
 182         if delta <= 0:
 183             continue # 0 is already process, <0 will be processed when nick swaps with nick2
 184         for nick2 in changes.keys():
 185             if changes[nick2]['delta'] == 0:
 186                 continue # already done
 187             if changes[nick2]['delta'] != -delta:
 188                 continue # not the good one
 189             percent = float(oldtdc[nick2] - newtdc[nick2]) / oldtdc[nick2] * 100.
 190             txtchanges.append('%s a pris %s cm² à %s (%.2f%%)' % (nick, number_format(delta), nick2, percent))
 191             changes[nick]['delta'] = 0
 192             changes[nick2]['delta'] = 0
 193
 194
 195     for nick, change in changes.items():
 196         delta = change['delta']
 197         if delta < 0:
 198             if nick in newtdc:
 199                 percent = float(-delta) / oldtdc[nick] * 100.
 200                 txtchanges.append('%s a perdu %s cm² (%.2f%%)' % (nick, number_format(-delta), percent))
 201             else:
 202                 txtchanges.append("%s a quité l'alliance avec %s cm²" % (nick, number_format(-delta)))
 203         elif delta > 0:
 204             if nick in oldtdc:
 205                 txtchanges.append('%s a gagné %s cm²' % (nick, number_format(delta)))
 206             else:
 207                 txtchanges.append("%s a rejoint l'alliance avec %s cm²" % (nick, number_format(delta)))
 208
 209     for txtchange in txtchanges:
 210         logging.info(txtchange)
 211
 212
 213 if __name__ == '__main__':
 214     from optparse import OptionParser
 215     parser = OptionParser()
 216     parser.add_option('-d', '--debug',
 217         action='store_true', dest='debug', default=False,
 218         help="debug mode")
 219     parser.add_option('-a', '--alliance',
 220         dest='alliance',
 221         help="alliance tag. default is to process player own alliance.")
 222     parser.add_option('--dry-run',
 223         action='store_true', dest='dryrun', default=False,
 224         help="don't store result in archives.")
 225     options, args = parser.parse_args()
 226
 227     if options.debug:
 228         loglevel = logging.DEBUG
 229     else:
 230         loglevel = logging.INFO
 231     logging.basicConfig(filename=RESULT, level=loglevel, format='%(asctime)s %(levelname)s %(message)s', datefmt='%Y-%m-%d %H:%M:%S %Z')
 232
 233     read_config()
 234
 235     oldtdc = None
 236     try:
 237         f = open(ARCHIVE, mode='r+', encoding='utf-8')
 238     except IOError as err:
 239         if err.errno == 2: # No such file or directory
 240             logging.warning("No archive file, creating one.")
 241             f = open(ARCHIVE, mode='w+', encoding='utf-8')
 242         else:
 243             raise
 244     else:
 245         oldtdc = eval(f.read())
 246
 247     newtdc = tdc_get(options.alliance)
 248     if oldtdc is not None:
 249         tdc_compare(oldtdc, newtdc)
 250
 251     # Save archive only after processing, just in case it crashes
 252     if not options.dryrun:
 253         f.seek(0)
 254         f.write(repr(newtdc))
 255         f.truncate()