-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathuploader.py
189 lines (154 loc) · 7.64 KB
/
uploader.py
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
#!/usr/bin/python
'''
Created on 6 Jun 2012
@author: Jeremy Blythe
Motion Uploader - uploads videos to Google Drive
Read the blog entry at http://jeremyblythe.blogspot.com for more information
Version 2: 10 Jun 2015 - rewritten for Google oauth2 and apiclient
'''
import sys
import smtplib
from datetime import datetime
import os.path
import logging
import httplib2
from apiclient import discovery
from oauth2client import client
from oauth2client.file import Storage
from googleapiclient.http import MediaFileUpload
import ConfigParser
class MotionUploader:
def __init__(self, config_file_path):
# Load config
config = ConfigParser.ConfigParser()
config.read(config_file_path)
# OAuth folder
self.oauth_folder = config.get('oauth', 'folder')
# GMail account credentials
self.username = config.get('gmail', 'user')
self.password = config.get('gmail', 'password')
self.from_name = config.get('gmail', 'name')
self.sender = config.get('gmail', 'sender')
# Recipient email address (could be same as from_addr)
self.recipient = config.get('gmail', 'recipient')
# Subject line for email
self.subject = config.get('gmail', 'subject')
# First line of email message
self.message = config.get('gmail', 'message')
# Folder (or collection) in Docs where you want the videos to go
self.folder = config.get('docs', 'folder')
# Folder (or collection) in Docs where you want the snapshot to go
self.snapshot_folder = config.get('docs', 'snapshot-folder')
# Options
self.delete_after_upload = config.getboolean('options', 'delete-after-upload')
self.send_email = config.getboolean('options', 'send-email')
self._create_drive()
def _create_drive(self):
"""Create a Drive service."""
auth_required = True
#Have we got some credentials already?
storage = Storage(self.oauth_folder+'uploader_credentials.txt')
credentials = storage.get()
try:
if credentials:
# Check for expiry
if credentials.access_token_expired:
if credentials.refresh_token is not None:
credentials.refresh(httplib2.Http())
auth_required = False
else:
auth_required = False
except:
# Something went wrong - try manual auth
pass
if auth_required:
flow = client.flow_from_clientsecrets(
self.oauth_folder+'client_secrets.json',
scope='https://www.googleapis.com/auth/drive',
redirect_uri='urn:ietf:wg:oauth:2.0:oob')
auth_uri = flow.step1_get_authorize_url()
print 'Go to this link in your browser:'
print auth_uri
auth_code = raw_input('Enter the auth code: ')
credentials = flow.step2_exchange(auth_code)
storage.put(credentials)
#Get the drive service
http_auth = credentials.authorize(httplib2.Http())
self.drive_service = discovery.build('drive', 'v2', http_auth)
def _get_folder_id(self, folder_name):
"""Find and return the id of the folder given the title."""
files = self.drive_service.files().list(q="title='%s' and mimeType contains 'application/vnd.google-apps.folder' and trashed=false" % folder_name).execute()
if len(files['items']) == 1:
folder_id = files['items'][0]['id']
return folder_id
else:
raise Exception('Could not find the %s folder' % folder_name)
def _send_email(self,msg):
'''Send an email using the GMail account.'''
senddate=datetime.strftime(datetime.now(), '%Y-%m-%d')
m="Date: %s\r\nFrom: %s <%s>\r\nTo: %s\r\nSubject: %s\r\nX-Mailer: My-Mail\r\n\r\n" % (senddate, self.from_name, self.sender, self.recipient, self.subject)
server = smtplib.SMTP('smtp.gmail.com:587')
server.starttls()
server.login(self.username, self.password)
server.sendmail(self.sender, self.recipient, m+msg)
server.quit()
def upload_video(self, video_file_path):
"""Upload a video to the specified folder. Then optionally send an email and optionally delete the local file."""
folder_id = self._get_folder_id(self.folder)
media = MediaFileUpload(video_file_path, mimetype='video/avi')
response = self.drive_service.files().insert(media_body=media, body={'title':os.path.basename(video_file_path), 'parents':[{u'id': folder_id}]}).execute()
#print response
video_link = response['alternateLink']
if self.send_email:
msg = self.message
if video_link:
msg += '\n\n' + video_link
self._send_email(msg)
if self.delete_after_upload:
os.remove(video_file_path)
def upload_snapshot(self, snapshot_file_path):
"""Upload a snapshot to the specified folder. Remove duplicates."""
folder_id = self._get_folder_id(self.snapshot_folder)
file_name = os.path.basename(snapshot_file_path)
#Delete the old snapshot
files = self.drive_service.files().list(q="title='%s' and '%s' in parents and trashed=false" % (file_name, folder_id)).execute()
if len(files['items']) >= 1:
for f in files['items']:
file_id = f['id']
self.drive_service.files().delete(fileId=file_id).execute()
#Now upload the new one
media = MediaFileUpload(snapshot_file_path, mimetype='image/jpeg')
self.drive_service.files().insert(media_body=media, body={'title':file_name, 'parents':[{u'id': folder_id}]}).execute()
def get_snapshot_url(self, snapshot_file_path):
"""Print out the public url for this snapshot."""
folder_id = self._get_folder_id(self.snapshot_folder)
public_url = 'https://googledrive.com/host/%s/' % folder_id
print public_url + os.path.basename(snapshot_file_path)
if __name__ == '__main__':
try:
logging.basicConfig(level=logging.ERROR)
if len(sys.argv) < 3:
exit('Motion Uploader - uploads videos and snapshots to Google Drive\n'+
' by Jeremy Blythe (http://jeremyblythe.blogspot.com)\n\n'+
' Usage: uploader.py {config-file-path} {source-file-path} [option]\n'+
' Options: snap : Upload a snapshot to the snapshot-folder\n'+
' snapurl : Print the public url for the folder+file\n'+
' None : Defaults to uploading video files')
cfg_path = sys.argv[1]
vid_path = sys.argv[2]
if len(sys.argv) > 3:
option = sys.argv[3]
else:
option = 'video'
if not os.path.exists(cfg_path):
exit('Config file does not exist [%s]' % cfg_path)
if not os.path.exists(vid_path):
exit('Source file does not exist [%s]' % vid_path)
if option.lower() == 'snap':
MotionUploader(cfg_path).upload_snapshot(vid_path)
elif option.lower() == 'snapurl':
MotionUploader(cfg_path).get_snapshot_url(vid_path)
else:
MotionUploader(cfg_path).upload_video(vid_path)
except Exception as e:
exit('Error: [%s]' % e)