diff --git a/server/download.py b/server/download.py index aadac12..fb56fb5 100644 --- a/server/download.py +++ b/server/download.py @@ -5,7 +5,17 @@ import paramiko import time paramiko.util.log_to_file('paramiko.log') +def fake_download(dest): + import requests, random + rnd = str(random.randint(1, 10000)) + r = requests.get('https://picsum.photos/400/300?random=' + rnd) + with open(dest / (rnd + '.jpg'), 'wb') as f: + f.write(r.content) + def download(ip, dest): + if settings.DEBUG: + fake_download(dest) + print('Downloading from', ip) port = 22 diff --git a/server/main.py b/server/main.py index 68913bf..489fc3d 100644 --- a/server/main.py +++ b/server/main.py @@ -26,7 +26,7 @@ def clients_post(): phone = str(content['phone']) for i in range(1, 100): - suffix = str(i).zfill(3) + suffix = str(i).zfill(2) folder = phone + '_' + suffix path = output_folder / folder if not path.exists(): @@ -99,11 +99,18 @@ def session_post(cid): # go through the photo taking process status = 'Warming up' - # warmup - power.lights_on() - time.sleep(4) - power.lights_off() - time.sleep(0.5) + try: + # warmup + power.lights_on() + time.sleep(4) + power.lights_off() + time.sleep(0.5) + 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) status = 'Capturing' diff --git a/server/power.py b/server/power.py index 5710e23..c3f265e 100644 --- a/server/power.py +++ b/server/power.py @@ -3,6 +3,8 @@ import settings import time def np_02B_api(ip, username, password, is_on): + if settings.DEBUG: return + if is_on: endpoint = '/cmd.cgi?grp=0' else: diff --git a/server/settings.py b/server/settings.py index 648a02c..fa558fc 100644 --- a/server/settings.py +++ b/server/settings.py @@ -1,3 +1,6 @@ +import os +DEBUG = os.environ.get('FLASK_ENV', None) == 'development' + LIGHT_IP = '192.168.99.25' LIGHT_USER = 'admin' LIGHT_PASS = 'admin'