def delete(id):
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute('delete from claim_params where claim = %s', (id))
cursor.execute('delete from claims where id = %s', (id))
return jsonify("OK")
def get():
request = "SELECT * FROM `garbage_types`"
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute(request)
result = cursor.fetchall()
return jsonify(result)
def add(args, collector_id, citizen_id):
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute(
"insert into claims (create_time, state, creator, executor) values (now(), 'new', %s, %s);",
(citizen_id, collector_id))
cursor.execute('SELECT LAST_INSERT_ID() as id FROM claims')
claim_id = cursor.fetchone()['id']
for key, value in args.items():
cursor.execute(
'insert into claim_params (claim, code, value) values (%s, %s, %s);',
(claim_id, key, value))
return jsonify({"id": claim_id})
def login(args):
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute(
'select id, `type` from users where login = %s and password = %s;',
(
args["login"],
hashlib.md5(args["password"].encode('utf-8')).hexdigest(),
))
role = cursor.fetchone()
if not role:
abort(401)
create_session(args["login"], role["type"])
return jsonify(dict(
id=role["id"],
role=role["type"],
))
def search(args):
conn = get_conn()
with conn.cursor() as cursor:
if 'collectorsId' in args:
cursor.execute(
"""select c.*, u.login as citizen, u2.login as collector
from claims c
inner join users u on u.id = c.creator
inner join users u2 on u2.id = c.executor
where c.executor = %s""", (args['collectorsId']))
elif 'citizensId' in args:
cursor.execute(
"""select c.*, u.login as collector, u2.login as citizen
from claims c
inner join users u on u.id = c.executor
inner join users u2 on u2.id = c.creator
where c.creator = %s""", (args['citizensId']))
else:
return jsonify(None)
result = cursor.fetchall()
for row in result:
row['create_time'] = row['create_time'].strftime(
'%Y-%m-%d %H:%M:%S')
row['params'] = {}
cursor.execute('select * from claim_params where claim = %s',
(row['id']))
res = cursor.fetchall()
for raw in res:
row['params'][raw['code']] = raw['value']
if 'garbage_type' in row['params']:
cursor.execute(
'select name from garbage_types where code = %s',
(row['params']['garbage_type']))
row['params']['garbage_type'] = cursor.fetchone()['name']
return jsonify(result)
def put(args, id, state):
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute('select * from claims where id = %s', (id))
result = cursor.fetchone()
result['params'] = {}
cursor.execute('select * from claim_params where claim = %s', (id))
res = cursor.fetchall()
for raw in res:
result['params'][raw['code']] = raw['value']
citizenId = result['creator']
collectorId = result['executor']
timestampTicks = int(time.time() * 10000000)
logger.info("RESULT: %r" % result)
if state == "done":
lotJson = json.dumps({
"LotType": result['params']['garbage_type'],
"Quantity": {
"MeasurementUnit": args["measurementUnit"],
"Value": args["amount"]
}
})
logger.info("JSON: %r" % lotJson)
cursor.execute(
'insert into events (CitizenId, CollectorId, TimestampTicks, LotJson) values (%s, %s, %s, %s);',
(citizenId, collectorId, timestampTicks, lotJson))
logger.info("INSERT EVENTS: %r" % args)
for key, value in args.items():
cursor.execute(
'insert ignore into claim_params (claim, code, value) values (%s, %s, %s);',
(id, key, value))
logger.info("END METHOD: %s" % state)
cursor.execute('update claims set state = %s where id = %s',
(state, id))
return jsonify("OK")
def search(args):
params, request = request_build(
garbage_type=args.get("garbageType", _sentinel))
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute(request, params)
collectors = cursor.fetchall()
@functools.lru_cache()
def get_params(user):
cursor.execute('select * from user_params where `user` = %s',
(user))
return cursor.fetchall()
for raw in collectors:
raw['params'] = {}
result = get_params(raw['user'])
for row in result:
raw['params'][row['code']] = row['value']
return jsonify(collectors)
def get(id):
conn = get_conn()
with conn.cursor() as cursor:
cursor.execute('select * from claims where id = %s', (id))
result = cursor.fetchone()
if not result:
return jsonify(None)
result['create_time'] = result['create_time'].strftime(
'%Y-%m-%d %H:%M:%S')
result['params'] = {}
cursor.execute('select * from claim_params where claim = %s', (id))
res = cursor.fetchall()
for raw in res:
result['params'][raw['code']] = raw['value']
if 'garbage_type' in result['params']:
cursor.execute('select name from garbage_types where code = %s',
(result['params']['garbage_type']))
result['params']['garbage_type'] = cursor.fetchone()['name']
return jsonify(result)