166 lines
3.6 KiB
Python
166 lines
3.6 KiB
Python
import os
|
|
import json
|
|
import time
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from flask import Flask, request, abort, send_from_directory
|
|
from flask_cors import CORS
|
|
|
|
import power, capture, download, settings
|
|
|
|
import logging
|
|
log = logging.getLogger('werkzeug')
|
|
if not settings.DEBUG:
|
|
log.setLevel(logging.ERROR)
|
|
|
|
build_folder = Path('../client/build')
|
|
output_folder = Path('./output')
|
|
app = Flask(__name__, static_folder=str(build_folder), static_url_path='')
|
|
CORS(app)
|
|
|
|
STANDBY = 0
|
|
WARMUP = 1
|
|
CAPTURING_PHOTO = 2
|
|
CAPTURING_GRID = 3
|
|
WRITING = 4
|
|
DOWNLOADING = 5
|
|
|
|
status = STANDBY
|
|
|
|
@app.route('/api/status', methods=['GET'])
|
|
def status_get():
|
|
return {'status': status}
|
|
|
|
@app.route('/api/clients', methods=['POST'])
|
|
def clients_post():
|
|
content = request.json
|
|
|
|
phone = str(content['phone'])
|
|
|
|
for i in range(1, 100):
|
|
suffix = str(i).zfill(2)
|
|
folder = phone + '_' + suffix
|
|
path = output_folder / folder
|
|
if not path.exists():
|
|
break
|
|
|
|
content['date'] = datetime.now().strftime('%Y-%m-%d')
|
|
content['time'] = datetime.now().strftime('%H:%M:%S')
|
|
|
|
path.mkdir()
|
|
info_file = path / 'info.txt'
|
|
info_file.touch()
|
|
info_file.write_text(json.dumps(content, indent=4))
|
|
|
|
client_id = folder
|
|
|
|
print('POST client:', content, 'cid:', client_id)
|
|
return {'client_id': client_id}
|
|
|
|
@app.route('/api/clients/<cid>', methods=['GET'])
|
|
def clients_get(cid):
|
|
folder = cid
|
|
path = output_folder / cid
|
|
|
|
if not path.exists():
|
|
abort(404)
|
|
|
|
info_file = path / 'info.txt'
|
|
info_text = info_file.read_text()
|
|
res = json.loads(info_text)
|
|
photo_glob = path.glob('*.jpg')
|
|
res['has_photos'] = bool(list(photo_glob))
|
|
|
|
print('GET client:', cid, 'res:', res)
|
|
return res
|
|
|
|
@app.route('/api/clients/<cid>/session', methods=['GET'])
|
|
def session_get(cid):
|
|
folder = cid
|
|
path = output_folder / cid
|
|
|
|
if not path.exists():
|
|
abort(404)
|
|
|
|
photo_glob = path.glob('*.jpg')
|
|
res = {}
|
|
res['photos'] = sorted([x.name for x in photo_glob])
|
|
|
|
return res
|
|
|
|
@app.route('/api/clients/<cid>/session', methods=['DELETE'])
|
|
def session_delete(cid):
|
|
folder = cid
|
|
path = output_folder / cid
|
|
|
|
if not path.exists():
|
|
abort(404)
|
|
|
|
photo_glob = path.glob('*.jpg')
|
|
|
|
for p in photo_glob:
|
|
p.unlink()
|
|
|
|
print('DELETE session:', cid)
|
|
return ''
|
|
|
|
@app.route('/api/clients/<cid>/session', methods=['POST'])
|
|
def session_post(cid):
|
|
content = request.json
|
|
light_time = content.get('light_time', 5000)
|
|
|
|
global status
|
|
|
|
print('POST session:', cid)
|
|
folder = cid
|
|
path = output_folder / cid
|
|
|
|
if not path.exists():
|
|
abort(404)
|
|
|
|
# go through the photo taking process
|
|
|
|
try:
|
|
# warmup
|
|
status = WARMUP
|
|
power.lights_on()
|
|
time.sleep(2)
|
|
power.lights_off()
|
|
time.sleep(1)
|
|
except BaseException as e:
|
|
print('Problem with lights: {} - {}'.format(e.__class__.__name__, str(e)))
|
|
print()
|
|
print('Are you sure the system is connected?')
|
|
print()
|
|
abort(500)
|
|
|
|
# capture
|
|
status = CAPTURING_PHOTO
|
|
power.lights_on()
|
|
time.sleep(0.1)
|
|
capture.trigger_capture()
|
|
time.sleep(light_time / 1000)
|
|
power.lights_off()
|
|
|
|
status = DOWNLOADING
|
|
download.download_all_photos(path)
|
|
time.sleep(3)
|
|
|
|
status = STANDBY
|
|
print('Finished.')
|
|
return ''
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return app.send_static_file('index.html')
|
|
|
|
@app.errorhandler(404)
|
|
def not_found(e):
|
|
return app.send_static_file('index.html')
|
|
|
|
@app.route('/output/<path:filename>')
|
|
def output(filename):
|
|
return send_from_directory('output/', filename)
|
|
|
|
app.run(host='0.0.0.0')
|