Jeena
1f6f2cd6be
Luckily in our case only the database name and the name of the directory where the files are saved have changed.
110 lines
3.5 KiB
Python
Executable file
110 lines
3.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import synology
|
|
import shutil
|
|
import os
|
|
import pickle
|
|
import json
|
|
from googleapiclient.discovery import build
|
|
from google.auth.transport.requests import Request
|
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
|
import google_auth_httplib2 # This gotta be installed for build() to work
|
|
from googleapiclient.http import MediaFileUpload
|
|
import requests
|
|
import subprocess
|
|
import shlex
|
|
|
|
def get_creds():
|
|
SCOPES = ['https://www.googleapis.com/auth/photoslibrary.appendonly',
|
|
'https://www.googleapis.com/auth/photoslibrary.sharing',
|
|
'https://www.googleapis.com/auth/photoslibrary',
|
|
'https://www.googleapis.com/auth/photoslibrary.readonly']
|
|
creds = None
|
|
if(os.path.exists("token.pickle")):
|
|
with open("token.pickle", "rb") as tokenFile:
|
|
creds = pickle.load(tokenFile)
|
|
if not creds or not creds.valid:
|
|
if (creds and creds.expired and creds.refresh_token):
|
|
creds.refresh(Request())
|
|
else:
|
|
# ssh -L 8081:localhost:8081 burgpreppach.swierczyniec.info
|
|
flow = InstalledAppFlow.from_client_secrets_file('client_secret.json', SCOPES)
|
|
creds = flow.run_local_server(port = 8081)
|
|
with open("token.pickle", "wb") as tokenFile:
|
|
pickle.dump(creds, tokenFile)
|
|
return creds
|
|
|
|
def login_gphoto(creds):
|
|
return build('photoslibrary', 'v1', credentials = creds, static_discovery=False)
|
|
|
|
def get_album(service, name):
|
|
results = service.albums().list(pageSize=10, fields='nextPageToken,albums(id,title)').execute()
|
|
if 'albums' in results:
|
|
for r in results['albums']:
|
|
if r['title'] == name:
|
|
return r
|
|
return None
|
|
|
|
def create_album(service, name):
|
|
request_body = {'album': {'title': name }}
|
|
results = service.albums().create(body=request_body).execute()
|
|
return results
|
|
|
|
def empty_album(service, album):
|
|
request_body = {'albumId': album['id'], 'pageSize': 100}
|
|
results = service.mediaItems().search(body=request_body).execute()
|
|
mediaItems = results.get('mediaItems')
|
|
if mediaItems == None:
|
|
return
|
|
mediaIds = []
|
|
for item in mediaItems:
|
|
mediaIds.append(item['id'])
|
|
|
|
request_body = { 'mediaItemIds': mediaIds }
|
|
response = service.albums().batchRemoveMediaItems(albumId=album['id'], body=request_body).execute()
|
|
|
|
def upload_gphoto(service, creds, album, picture_path):
|
|
upload_url = 'https://photoslibrary.googleapis.com/v1/uploads'
|
|
headers = {
|
|
'Authorization': 'Bearer ' + creds.token,
|
|
'Content-type': 'application/octet-stream',
|
|
'X-Goog-Upload-Protocol': 'raw'
|
|
}
|
|
img = open(os.path.join(picture_path), 'rb').read()
|
|
response = requests.post(upload_url, data=img, headers=headers)
|
|
request_body = {
|
|
'albumId': album['id'],
|
|
'newMediaItems': [
|
|
{
|
|
'simpleMediaItem': {
|
|
'uploadToken': response.content.decode('utf-8')
|
|
}
|
|
}
|
|
]
|
|
}
|
|
upload_response = service.mediaItems().batchCreate(body=request_body).execute()
|
|
|
|
if __name__ == "__main__":
|
|
|
|
get_pictures = True
|
|
if get_pictures:
|
|
host_ip = os.getenv('DB_HOST')
|
|
conn = synology.connect_db(host_ip, os.getenv('DB_USER'), os.getenv('DB_PASSWD'))
|
|
names = "richard|yingfen"
|
|
pictures = synology.fetch_paths_for_names(conn, names, 20)
|
|
synology.close_db(conn)
|
|
dirpath = synology.fetch_files("jeena@" + host_ip, "/var/services/homes/jeena/Photos", pictures)
|
|
|
|
creds = get_creds()
|
|
gclient = login_gphoto(creds)
|
|
album = get_album(gclient, "Synology")
|
|
if album == None:
|
|
album = create_album(gclient, "Synology")
|
|
|
|
empty_album(gclient, album)
|
|
|
|
with os.scandir(dirpath) as dirs:
|
|
for entry in dirs:
|
|
upload_gphoto(gclient, creds, album, os.path.join(dirpath, entry.name))
|
|
|
|
shutil.rmtree(dirpath)
|