From d1f527c93b59ff39e29fbd3895569d31a0bfd533 Mon Sep 17 00:00:00 2001 From: Tanner Collin Date: Sun, 7 Mar 2021 18:53:06 -0700 Subject: [PATCH] Add module to download photos --- server/.gitignore | 2 + server/download.py | 42 +++++++++++++++++++++ server/settings.py | 94 ++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 138 insertions(+) create mode 100644 server/download.py diff --git a/server/.gitignore b/server/.gitignore index 4a8da09..947f30e 100644 --- a/server/.gitignore +++ b/server/.gitignore @@ -101,3 +101,5 @@ ENV/ # Editor *.swp *.swo + +paramiko.log diff --git a/server/download.py b/server/download.py new file mode 100644 index 0000000..5ad784b --- /dev/null +++ b/server/download.py @@ -0,0 +1,42 @@ +import os +import settings +import threading +import paramiko +import time +paramiko.util.log_to_file('paramiko.log') + +def download(ip, dest): + print('Downloading from', ip) + + port = 22 + transport = paramiko.Transport((ip, port)) + transport.connect(None, settings.RASPBERRY_USER, settings.RASPBERRY_PASS) + + sftp = paramiko.SFTPClient.from_transport(transport) + + files = sftp.listdir('/3dscan/') + + for f in files: + source_file = '/3dscan/' + f + dest_file = dest + f + print('Grabbing file', source_file) + sftp.get(source_file, dest_file) + sftp.remove(source_file) + + if sftp: sftp.close() + if transport: transport.close() + + print('Finished downloading from', ip) + +def download_all_photos(dest): + if not dest.endswith('/'): + raise Exception('Destination must end with /') + + if not os.path.exists(dest): + raise Exception('Destination does not exist') + + print('Downloading all photos to', dest) + + for ip in settings.RASPBERRY_IPS: + t = threading.Thread(target=download, args=(ip, dest)) + t.start() diff --git a/server/settings.py b/server/settings.py index d180801..648a02c 100644 --- a/server/settings.py +++ b/server/settings.py @@ -5,3 +5,97 @@ LIGHT_PASS = 'admin' GRID_IP = '192.168.99.20' GRID_USER = 'admin' GRID_PASS = 'admin' + +RASPBERRY_USER = 'root' +RASPBERRY_PASS = '3dscan' +RASPBERRY_IPS = [ + '192.168.99.101', + '192.168.99.103', + '192.168.99.104', + '192.168.99.105', + '192.168.99.106', + '192.168.99.107', + '192.168.99.108', + '192.168.99.109', + '192.168.99.110', + '192.168.99.111', + '192.168.99.112', + '192.168.99.113', + '192.168.99.114', + '192.168.99.115', + '192.168.99.116', + '192.168.99.117', + '192.168.99.118', + '192.168.99.119', + '192.168.99.120', + '192.168.99.121', + '192.168.99.122', + '192.168.99.123', + '192.168.99.124', + '192.168.99.125', + '192.168.99.126', + '192.168.99.127', + '192.168.99.128', + '192.168.99.129', + '192.168.99.130', + '192.168.99.131', + '192.168.99.132', + '192.168.99.133', + '192.168.99.134', + '192.168.99.135', + '192.168.99.136', + '192.168.99.137', + '192.168.99.138', + '192.168.99.139', + '192.168.99.140', + '192.168.99.141', + '192.168.99.142', + '192.168.99.143', + '192.168.99.144', + '192.168.99.145', + '192.168.99.146', + '192.168.99.147', + '192.168.99.148', + '192.168.99.149', + '192.168.99.150', + '192.168.99.151', + '192.168.99.152', + '192.168.99.153', + '192.168.99.154', + '192.168.99.155', + '192.168.99.156', + '192.168.99.157', + '192.168.99.158', + '192.168.99.159', + '192.168.99.160', + '192.168.99.161', + '192.168.99.162', + '192.168.99.163', + '192.168.99.164', + '192.168.99.165', + '192.168.99.166', + '192.168.99.167', + '192.168.99.168', + '192.168.99.169', + '192.168.99.170', + '192.168.99.171', + '192.168.99.172', + '192.168.99.173', + '192.168.99.174', + '192.168.99.175', + '192.168.99.176', + '192.168.99.177', + '192.168.99.178', + '192.168.99.179', + '192.168.99.180', + '192.168.99.181', + '192.168.99.182', + '192.168.99.183', + '192.168.99.184', + '192.168.99.185', + '192.168.99.191', + '192.168.99.192', + '192.168.99.193', + '192.168.99.194', + '192.168.99.195', +]