gdrive-python/drive.py at master · archmaester/gdrive-python · 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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
import os
import argparse
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive
from os import chdir, listdir, stat

def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(
        description='API for managing Google drive')
    parser.add_argument(
        '--upload-file', dest='upload_file',
        help='Uploading file to Google Drive',
        action='store_true')
    parser.add_argument(
        '--download-file', dest='download_file',
        help='Downloading file from Google Drive',
        action='store_true')
    parser.add_argument(
        '--upload-folder', dest='upload_folder',
        help='Uploading folder to Google Drive',
        action='store_true')
    parser.add_argument(
        '--parent-id', dest='parent_id', help='Id of the parent folder',
        default='', type=str)
    parser.add_argument(
        '--file-id', dest='file_id', help='Id of the file',
        default='', type=str)
    parser.add_argument(
        '--file-name', dest='file_name',
        help='title of the file that will be saved',
        default='', type=str)
    parser.add_argument(
        '--folder-name', dest='folder_name',
        help='title of the file that will be saved',
        default='', type=str)
    parser.add_argument(
        '--upload-name', dest='upload_name',
        help='title of the file that will be saved',
        default='', type=str)
    parser.add_argument(
        '--download-name', dest='download_name',
        help='title of the file that will be saved',
        default='', type=str)
    parser.add_argument(
        '--file-path', dest='file_path',
        help='Path of the file where it is to be saved/fetched',
        default='', type=str)
    parser.add_argument(
        '--folder-path', dest='folder_path',
        help='Path of the file where it is to be saved/fetched',
        default='', type=str)

    args = parser.parse_args()
    return args


def main():

    args = parse_args()
    gauth = GoogleAuth()
    gauth.LoadClientConfigFile(os.environ["DRIVE_SECRET_PATH"])
    drive = GoogleDrive(gauth)

    if args.upload_file:

        file = drive.CreateFile({"title": args.upload_name,
                                 'parents': [{'id': args.parent_id}]})
        file.SetContentFile(os.path.join(args.file_path, args.file_name))
        file.Upload()
        print("You have successfully uploaded the file")

    if args.upload_folder:

        create_folder(drive, args.folder_name,
                      args.parent_id, args.folder_path)

    if args.download_file:
        download_file(drive, args.file_id, args.download_name, args.file_path)

def create_folder(drive, folder_name, parent_id, folder_path):

    folder_metadata = {
        'title': folder_name,
        'mimeType': 'application/vnd.google-apps.folder',
        'parents': [{"kind": "drive#fileLink", "id": parent_id}]
    }

    folder = drive.CreateFile(folder_metadata)
    folder.Upload()
    upload_files(drive, folder['id'], folder_path)


def upload_files(drive, folder_id, folder_path):

    for file in os.listdir(folder_path):

        statinfo = stat(os.path.join(folder_path, file))
        if statinfo.st_size > 0:

            print('Uploading .......' + file)
            if os.path.isdir(os.path.join(folder_path, file)):
                create_folder(drive, file, folder_id, os.path.join(folder_path, file))
            else:
                f = drive.CreateFile({
                    "title": file,
                    "parents": [{"kind": "drive#fileLink", "id": folder_id}]})
                f.SetContentFile(os.path.join(folder_path, file))
                f.Upload()
            print('Uploaded :):) ' + file)
        else:
            print('File {0} is empty :('.format(file))


def download_folder(drive, folder_id, savel_path):

    pass


def download_file(drive, file_id, save_name, save_path):

    file = drive.CreateFile({'id': file_id})
    file.GetContentFile(os.path.join(save_path, file['title']))


if __name__ == "__main__":
    main()