google-API/tagger.py at github · labroid/google-API · 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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
__author__ = 'scott_jackson'
import httplib2
import os
import json
import arrow
import pymongo
import sys

from apiclient import discovery
import oauth2client
from oauth2client import client
from oauth2client import tools

try:
    import argparse
    flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
except ImportError:
    flags = None

SCOPES = 'https://www.googleapis.com/auth/drive.metadata'
CLIENT_SECRET_FILE = 'client_secret.json'
APPLICATION_NAME = 'Other Client 1'


def get_credentials():
    """Gets valid user credentials from storage.

    If nothing has been stored, or if the stored credentials are invalid,
    the OAuth2 flow is completed to obtain the new credentials.

    Returns:
        Credentials, the obtained credential.
    """
    home_dir = os.path.expanduser('~')
    credential_dir = os.path.join(home_dir, '.credentials')
    if not os.path.exists(credential_dir):
        os.makedirs(credential_dir)
    credential_path = os.path.join(credential_dir, 'drive-tagger.json')
    store = oauth2client.file.Storage(credential_path)
    credentials = store.get()
    if not credentials or credentials.invalid:
        flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
        flow.user_agent = APPLICATION_NAME
        if flags:
            credentials = tools.run_flow(flow, store, flags)
        else: # Needed only for compatability with Python 2.6
            credentials = tools.run(flow, store)
        print('Storing credentials to ' + credential_path)
    return credentials

def _safe_print(u, errors="replace"):
    """Safely print the given string.

    If you want to see the code points for unprintable characters then you
    can use `errors="xmlcharrefreplace"`.
    """
    s = u.encode(sys.stdout.encoding or "utf-8", errors)
    print(s)

def main():
    credentials = get_credentials()
    http = credentials.authorize(httplib2.Http())
    service = discovery.build('drive', 'v2', http=http)

    file_list = None
    traversed = 0
    added = 0
    skipped = 0
    start = arrow.now()
    while True:
        if file_list is None:
            file_list = service.files().list(q="(mimeType contains 'image/' or mimeType contains 'video/')", maxResults=1, fields="items/id, nextPageToken").execute()
        else:
            if file_list['nextPageToken'] is None:
                break
            file_list = service.files().list(pageToken=file_list['nextPageToken'], maxResults=1000, fields="items/id, nextPageToken").execute()

        for f in file_list['items']:
            traversed += 1
            file_id = '1el16TSNYvaGQndXXDZjheN_CANnWbA9wSA'
#            md = service.files().get(fileId=f['id'], fields="id, md5Checksum, properties").execute()
#            md = service.files().get(fileId=f['id']).execute()
            md = service.files().get(fileId = file_id).execute()
            print "files().get() returns:"
            print json.dumps(md, indent=4)
            md = service.properties().list(fileId=file_id).execute()
            print "properties().list() returns:"
            print json.dumps(md, indent=4)
            # update = {
            #     'properties': [
            #         {
            #             "key": "md5sum",
            #             "value:": md['md5Checksum']
            #         }
            #     ]
            # }
     #       foo = service.files().patch(fileId=md['id'],body=update).execute()
     #       print type(foo)
     #       print json.dumps(foo, indent=4)
            break
        break


        #         if 'imageMediaMetadata' in md:
        #             md['date'] = md['imageMediaMetadata']['date']
        #             del md['imageMediaMetadata']
        #         else:
        #             md['date'] = None
        #         _safe_print(md['title'])
        #         db.insert(md)
        #         added += 1
        #     else:
        #         #print "Id already in database"
        #         skipped += 1
        #
        # print "Traversed: {}, Added: {}, Skipped: {}, Elapsed time: {}".format(traversed, added, skipped, arrow.now() - start)

    # tree_root = service.children().list(folderId='root', q="title = 'Google Photos'", fields="items/id").execute()
    # print "Tree Root:", tree_root
    # top_level = service.children().list(folderId=tree_root['items'][0]['id'], fields="items/id").execute()
    # print "Top Level:", json.dumps(top_level['items'])
    # for d in top_level['items']:
    #     dstats = service.files().get(fileId=d['id'], fields="mimeType, id, title").execute()
    #     print dstats


    print "***Done***"

    # results = service.files().list(q="title = 'IMG_1773.JPG'", maxResults=10).execute()
    # print type(results)
    # print results
    # print json.dumps(results)

#    results = service.files().list().execute()
#    items = results.get('items', [])

#    print type(items)
#    print "Length=",len(items)
    # if not items:
    #     print('No files found.')
    # else:
    #     print('Files:')
    #     for item in items:
    #         print('{0} ({1})'.format(item['title'], item['id']))

if __name__ == '__main__':
    main()