cloud_project_desktop_client/dir_handler.py at master · bilgincan/cloud_project_desktop_client · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import socket
import os
import tqdm
from encryption import *
# from client_handler import login_procedure

BUFFER_SIZE = 4096

def receive_files_and_write(s, path, key):
    try:
        os.mkdir(path)
    except OSError:
        print(OSError)

    l = 1
    file_name = ""
    f = ""
    while(l):
        l = s.recv(BUFFER_SIZE)
        l = decrypt(key, l)
        print(path)
        try:
            msg = l
            if "mkdirNcd" in msg:
                new_dir = msg.split("mkdirNcd ")[1]
                subdir = path +"/"+ new_dir
                receive_files_and_write(s, subdir, key)
                continue
            elif msg == "D6642D2FE679ED121CB9067EE0DC10C244CCAB926BCCD9D18C3021E66330384C":
                file_name = s.recv(BUFFER_SIZE)
                file_name = decrypt(key, file_name)
                f = open((path+"/"+file_name),"w+")
                print(file_name)
                print("new plain text")
                continue
            elif "cd .." in msg:
                return
            if f != "":
                f.write(msg)
        except (UnicodeDecodeError, TypeError):
            if f != "":
                f.close()
            f = open((path+"/"+file_name), "wb+")
            print("new binary file")
            while(l):
                if not f.closed:
                    f.write(l)
                l = s.recv(BUFFER_SIZE)
                l = decrypt(key, l)
                try:
                    msg = l
                    if msg == "db2350b264d0b0a4b14528c94dbfaf7ef7cc90a525062bb39f5eba3093e9e095":
                        break
                    if msg == "D6642D2FE679ED121CB9067EE0DC10C244CCAB926BCCD9D18C3021E66330384C":
                        f.close()
                        file_name = s.recv(BUFFER_SIZE)
                        file_name = decrypt(key, file_name)
                        print(file_name)
                        f = open((path+"/"+file_name),"w+")
                        break
                    elif "cd .." in msg:
                        return
                    elif "mkdirNcd" in msg:
                        new_dir = msg.split("mkdirNcd ")[1]
                        subdir = path +"/"+ new_dir
                        receive_files_and_write(s, subdir, key)
                        continue
                except UnicodeDecodeError:
                    continue
        except IsADirectoryError:
            s.close()
            return 1

        print("aldı")
    #s.send("received".encode())
    s.close()
    print("socket has been closed")
    if f != "":
        f.close()
    return 0