52 lines
1.1 KiB
Python
Executable file
52 lines
1.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
"""Handle API requests to the database
|
|
something
|
|
"""
|
|
import json
|
|
import sqlite3
|
|
|
|
|
|
DATABASE = '/usr/local/www/mocha-server/mocha.db'
|
|
|
|
|
|
def fetch(user_id):
|
|
conn = sqlite3.connect(DATABASE)
|
|
conn.row_factory = sqlite3.Row
|
|
cursor = conn.cursor()
|
|
|
|
if user_id != '*':
|
|
cursor.execute("SELECT * FROM users WHERE user_id=?", user_id)
|
|
else:
|
|
cursor.execute("SELECT * FROM users")
|
|
|
|
output = cursor.fetchall()
|
|
output = json.dumps([dict(row) for row in output])
|
|
conn.close()
|
|
|
|
if output is None:
|
|
output = ('Error',)
|
|
return output
|
|
|
|
|
|
def process_request(uri):
|
|
parts = uri.split('/')[1:]
|
|
assert parts[0] == 'mocha'
|
|
|
|
output = ''
|
|
|
|
if parts[1] == 'users':
|
|
output = fetch(parts[2])
|
|
return output
|
|
|
|
print(parts)
|
|
|
|
|
|
def application(environ, start_response):
|
|
status = '200 OK'
|
|
output = process_request(environ['REQUEST_URI']).encode('UTF-8')
|
|
|
|
response_headers = [('Content-type', 'application/json'),
|
|
('Content-Length', str(len(output)))]
|
|
start_response(status, response_headers)
|
|
|
|
return [output]
|