233 lines
7.3 KiB
Python
233 lines
7.3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# lunchuman.wsgi: a WSGI script for managing a simple lunch club
|
|
# Copyright (C) 2012 Niels G. W. Serup
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public
|
|
# License along with this program. If not, see
|
|
# <http://www.gnu.org/licenses/>.
|
|
|
|
## Version: 0.1.0
|
|
## Maintainer: Niels G. W. Serup <ns@metanohi.name>
|
|
## Temporary website: http://datalosjen.metanohi.name/static/source/lunchuman
|
|
|
|
import sys
|
|
import os
|
|
import bottle
|
|
from bottle import route, abort, error, request, redirect
|
|
import sqlite3
|
|
import datetime
|
|
import threading
|
|
import itertools
|
|
import atexit
|
|
import pickle
|
|
import configparser
|
|
import locale
|
|
import traceback
|
|
|
|
_preferred_encoding = 'utf-8' # locale.getpreferredencoding() # fail
|
|
_absfile = os.path.abspath(__file__)
|
|
_filedir = os.path.dirname(_absfile)
|
|
_configfile_name = 'lunchuman.config'
|
|
|
|
def start_bottle():
|
|
global application
|
|
application = bottle.default_app()
|
|
|
|
def block():
|
|
threading.Event().wait()
|
|
|
|
with open(os.path.join(_filedir, '.dynamicsettings'), 'rb') as f:
|
|
dynamic_settings = pickle.load(f)
|
|
|
|
if dynamic_settings['status'] == 'stop':
|
|
start_bottle()
|
|
block()
|
|
|
|
|
|
class LunchumanDatabase:
|
|
def __init__(self, fname):
|
|
self.dbfile = fname
|
|
|
|
def load(self):
|
|
self.conn = sqlite3.connect(self.dbfile)
|
|
self.cursor = self.conn.cursor()
|
|
self._create()
|
|
|
|
def close(self):
|
|
self.conn.close()
|
|
|
|
def _create(self):
|
|
self.cursor.execute('''
|
|
create table if not exists users (
|
|
id integer primary key autoincrement,
|
|
name text,
|
|
moneys int,
|
|
foods int
|
|
)
|
|
''')
|
|
self.cursor.execute('''
|
|
create table if not exists actions (
|
|
id integer primary key autoincrement,
|
|
action_when datetime,
|
|
action text
|
|
)
|
|
''')
|
|
self.conn.commit()
|
|
|
|
def get_users(self):
|
|
self.cursor.execute('select * from users order by moneys / (foods * 1.0) desc')
|
|
return self.cursor.fetchall()
|
|
|
|
def add_user(self, name):
|
|
self.cursor.execute('insert into users values (NULL, ?, 0, 0)', (name,))
|
|
self.log('new user created: {}'.format(repr(name)))
|
|
self.conn.commit()
|
|
|
|
def log(self, msg):
|
|
self.cursor.execute('insert into actions values (NULL, ?, ?)',
|
|
(datetime.datetime.utcnow(), msg))
|
|
self.conn.commit()
|
|
|
|
def get_log(self):
|
|
self.cursor.execute('select * from actions order by action_when desc')
|
|
return self.cursor.fetchall()
|
|
|
|
def update_user_data(self, *users):
|
|
log_msg = []
|
|
for id, new_name, moneys_incr, foods_incr in users:
|
|
try:
|
|
foods_incr = int(foods_incr)
|
|
except ValueError:
|
|
foods_incr = 0
|
|
try:
|
|
moneys_incr = int(moneys_incr)
|
|
except ValueError:
|
|
moneys_incr = 0
|
|
self.cursor.execute('select name from users where id=?', (id,))
|
|
old_name = self.cursor.fetchone()[0]
|
|
self.cursor.execute('''
|
|
update users set name=?, moneys=moneys + (?), foods=foods + (?)
|
|
where id=?
|
|
''', (new_name, moneys_incr, foods_incr, id))
|
|
loclog = []
|
|
if old_name != new_name:
|
|
loclog.append('change name to {}'.format(new_name))
|
|
if foods_incr > 0:
|
|
loclog.append('eat {} foods'.format(foods_incr))
|
|
elif foods_incr < 0:
|
|
loclog.append('uneat {} foods'.format(-foods_incr))
|
|
if moneys_incr > 0:
|
|
loclog.append('spend {} moneys'.format(moneys_incr))
|
|
elif moneys_incr < 0:
|
|
loclog.append('unspend {} moneys'.format(-moneys_incr))
|
|
if loclog:
|
|
log_msg.append('{} do:\n'.format(old_name)
|
|
+ ' ' + '\n '.join(loclog))
|
|
if log_msg:
|
|
self.log('\n'.join(log_msg))
|
|
self.conn.commit()
|
|
|
|
def add_settings_from_file(fname, settings):
|
|
cfg = configparser.SafeConfigParser(settings)
|
|
cfg.add_section('lunchuman')
|
|
cfg.read(os.path.join(_filedir, fname))
|
|
return {name: value for name, value in cfg.items('lunchuman')}
|
|
|
|
def dictpair(*ds):
|
|
return {key: val for key, val
|
|
in itertools.chain(*(d.items() for d in ds))}
|
|
|
|
def read_file(locpath):
|
|
with open(os.path.join(_filedir, locpath), 'rb') as f:
|
|
return f.read().decode(_preferred_encoding)
|
|
|
|
def get_seq_input(d, base_key):
|
|
i = 0
|
|
xs = []
|
|
while True:
|
|
try:
|
|
xs.append(d['{}[{}]'.format(base_key, i)])
|
|
except KeyError:
|
|
break
|
|
i += 1
|
|
return xs
|
|
|
|
settings = add_settings_from_file(
|
|
_configfile_name, {
|
|
'name': 'Lunchuman',
|
|
'dbfile': 'lunchuman.sqlite',
|
|
'template': 'template.html',
|
|
'frontpage': 'frontpage.html'
|
|
})
|
|
|
|
_template = read_file(settings['template'])
|
|
_userdata_template = '''
|
|
<input type='hidden' name='id[{i}]' value='{id}' />
|
|
<input type='hidden' name='orig_name[{i}]' value={name} />
|
|
<tr>
|
|
<td><input type='text' name='new_name[{i}]' style='width: 140px;'
|
|
value={name} /></td>
|
|
<td>{moneys} + <input type='text' name='moneys_incr[{i}]'
|
|
style='width: 40px;' value='0' class='deleteonfocus' /></td>
|
|
<td>{foods} + <input type='text' name='foods_incr[{i}]'
|
|
style='width: 40px;' value='0' class='deleteonfocus' /></td>
|
|
<td>{goodwill}</td>
|
|
</tr>
|
|
'''
|
|
_frontpage = _template.format(**dictpair(settings, {
|
|
'content': read_file(settings['frontpage']),
|
|
'headextra': "<script type='text/javascript' \
|
|
src='/static/front.js'></script>"
|
|
}))
|
|
|
|
db = LunchumanDatabase(os.path.join(_filedir, settings['dbfile']))
|
|
#atexit.register(db.close)
|
|
db.load()
|
|
|
|
|
|
@route('/')
|
|
def frontpage():
|
|
users = db.get_users()
|
|
userdata = '\n'.join(_userdata_template.format(
|
|
id=users[i][0], name=repr(users[i][1]), moneys=users[i][2],
|
|
foods=users[i][3], i=i, goodwill=('{:.2f}'.format(
|
|
users[i][2] / users[i][3]) if users[i][3] > 0
|
|
else None))
|
|
for i in range(len(users)))
|
|
log = '\n\n'.join('{} UTC:\n{}'.format(*row[1:]) for row in db.get_log())
|
|
return _frontpage.format(userdata=userdata, log=log)
|
|
|
|
@route('/adduser', method='post')
|
|
def add_user():
|
|
db.add_user(request.forms['name'])
|
|
redirect('/')
|
|
|
|
@route('/updateuserdata', method='post')
|
|
def update_user_data():
|
|
try:
|
|
users = (vals[1:] for vals in
|
|
filter(lambda vals: vals[0] != vals[2] or
|
|
(vals[3] != '0' and vals[3] != '') or
|
|
(vals[4] != '0' and vals[4] != ''),
|
|
zip(*(get_seq_input(request.forms, attr)
|
|
for attr in ('orig_name', 'id', 'new_name',
|
|
'moneys_incr', 'foods_incr')))))
|
|
db.update_user_data(*users)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
redirect('/')
|
|
|
|
|
|
start_bottle()
|