wetterturnier-backend/ComputePetrus.py at master · sferics/wetterturnier-backend · GitHub

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
# -------------------------------------------------------------------
# - NAME:        ComputePetrus.py
# - AUTHOR:      Reto Stauffer
# - DATE:        2014-09-19
# - LICENSE: GPL-3, Reto Stauffer, copyright 2014
# -------------------------------------------------------------------
# - DESCRIPTION: Compute Petrus for the next tournament date.
#                Can be started several times.
# -------------------------------------------------------------------
# - EDITORIAL:   2014-09-13, RS: Created file on thinkreto.
# -------------------------------------------------------------------
# - L@ST MODIFIED: 2018-01-07 14:45 on marvin
# -------------------------------------------------------------------


# - Start as main script (not as module)
if __name__ == '__main__':

   import sys, os
   import numpy as np
   # - Wetterturnier specific modules
   from pywetterturnier import utils
   from pywetterturnier import database
   from pywetterturnier import mitteltip

   # - Evaluating input arguments
   inputs = utils.inputcheck('ComputePetrus')
   # - Read configuration file
   config = utils.readconfig('config.conf',inputs)

   # - If input_user was given as string we have to find the
   #   corresponding userID first!
   if not config['input_user'] == None:
      print '[!] NOTE: got input -u/--user. Will be ignored in ComputePetrus.'
      config['input_user'] = None


   # - Initializing class and open database connection
   db        = database.database(config)
   # - Loading tdate (day since 1970-01-01) for the tournament.
   #   Normaly Friday-Tornament (tdate is then Friday) while
   #   the bet-dates are for Saturday and Sunday if there was
   #   no input tournament date -t/--tdate.
   if config['input_tdate'] == None:
      tdate      = db.current_tournament()
   else:
      tdate      = config['input_tdate']

   print '  * Current tournament is %s' % utils.tdate2string( tdate )
   # - Loading all different cities (active cities)
   cities     = db.get_cities()
   # - If input city set, then drop all other cities.
   if not config['input_city'] == None:
      tmp = []
      for elem in cities:
         if elem['name'] == config['input_city']: tmp.append( elem )
      cities = tmp

   # ----------------------------------------------------------------
   # - Check if we are allowed to perform the computation of the
   #   Petrus bets on this date
   # ----------------------------------------------------------------
   check = utils.datelock(config,tdate)
   if check:
      print '    Date is \'locked\' (datelock). Dont execute.'
      import sys; sys.exit(0)


   # ----------------------------------------------------------------
   # - Prepare the Petrus
   # ----------------------------------------------------------------
   username = 'Petrus'
   db.create_user( username )
   userID = db.get_user_id( username )

   # ----------------------------------------------------------------
   # - Compute its mitteltip, one for each station.
   # ----------------------------------------------------------------
   for city in cities:

      print '\n  * Compute the %s for city %s (ID: %d)' % (username,city['name'], city['ID'])

      # - Returns list object containing two dicts
      #   where all the bets are in.
      bet = mitteltip.mitteltip(db,'all',False,city,tdate)

      # - If bet is False, continue
      if bet == False: continue

      # -------------------------------------------------------------
      # - Inserting into database now
      # -------------------------------------------------------------
      print '    Inserting data into database now'
      for day in range(1,3):
         for k in bet[day-1].keys():
            paramID = db.get_parameter_id(k)
            db.upsert_bet_data(userID,city['ID'],paramID,tdate,day,bet[day-1][k])

   db.commit()
   db.close()