Upload New File

begin.py

+import socket
+import os
+import sys
+import struct
+#import time
+
+ip = '192.168.191.4'
+
+try:
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.connect((ip,8081))
+except socket.error as msg:
+    print(msg)
+    sys.exit(1)
+
+time_max = input("please enter the maximum time (seconds) allowed for picture(s): ")
+if time_max.isdigit():
+    s.send(time_max.encode('utf-8'))
+    s.close()
+
+    try:
+        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        s.connect((ip,8081))
+    except socket.error as msg:
+        print(msg)
+        sys.exit(1)
+
+    img_type = ['.jpg','.JPG','.png','.PNG','.bmp','.BMP']
+
+    while 1:
+        filepath = input("please input file or folder path: ")
+        if os.path.isdir(filepath):  
+            img_path_vec = [os.path.join(filepath,imgpath) for imgpath in os.listdir(filepath) if os.path.splitext(imgpath)[1] in img_type]
+            for img_name in img_path_vec:
+                fileinfo_size = struct.calcsize('128sq')
+                fhead = struct.pack('128sq', bytes(os.path.basename(img_name).encode('utf-8')),os.stat(img_name).st_size)
+                s.send(fhead)
+                print ('client filepath: {0}'.format(img_name))
+                fp = open(img_name, 'rb')
+                while 1:
+                    data = fp.read(1024)
+                    if not data:
+                        print ('{0} file send over...'.format(img_name))
+                        break
+                    s.send(data)
+                if img_name == img_path_vec[-1]:
+                    break
+                s.close()
+#                time.sleep(0.5)
+                try:
+                    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                    s.connect((ip,8081))
+                except socket.error as msg:
+                    print(msg)
+                    sys.exit(1)
+            s.close()
+            break
+        else:
+            if os.path.isfile(filepath):
+                if os.path.splitext(filepath)[1] in img_type:
+                    fileinfo_size = struct.calcsize('128sq')
+                    fhead = struct.pack('128sq', bytes(os.path.basename(filepath).encode('utf-8')),os.stat(filepath).st_size)
+                    s.send(fhead)
+                    print ('client filepath: {0}'.format(filepath))
+                    fp = open(filepath, 'rb')
+                    while 1:
+                        data = fp.read(1024)
+                        if not data:
+                            print ('{0} file send over...'.format(filepath))
+                            break
+                        s.send(data)
+                else:
+                    print('The file sent is not a picture.')
+            else:
+                print('Please enter the correct file or folder path.')
+            s.close()
+            break
+else:
+    print('Please key in numbers.')
\ No newline at end of file