151 lines
4.0 KiB
Python
151 lines
4.0 KiB
Python
import io
|
|
import json
|
|
import os
|
|
from datetime import datetime
|
|
|
|
import requests
|
|
from cookbook.models import Recipe, RecipeImport, SyncLog
|
|
from cookbook.provider.provider import Provider
|
|
|
|
|
|
class Dropbox(Provider):
|
|
|
|
@staticmethod
|
|
def import_all(monitor):
|
|
url = "https://api.dropboxapi.com/2/files/list_folder"
|
|
|
|
headers = {
|
|
"Authorization": "Bearer " + monitor.storage.token,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"path": monitor.path
|
|
}
|
|
|
|
r = requests.post(url, headers=headers, data=json.dumps(data))
|
|
try:
|
|
recipes = r.json()
|
|
except ValueError:
|
|
log_entry = SyncLog(status='ERROR', msg=str(r), sync=monitor)
|
|
log_entry.save()
|
|
return r
|
|
|
|
import_count = 0
|
|
# TODO check if has_more is set and import that as well
|
|
for recipe in recipes['entries']:
|
|
path = recipe['path_lower']
|
|
if not Recipe.objects.filter(file_path__iexact=path).exists() \
|
|
and not RecipeImport.objects.filter(file_path=path).exists(): # noqa: E501
|
|
name = os.path.splitext(recipe['name'])[0]
|
|
new_recipe = RecipeImport(
|
|
name=name,
|
|
file_path=path,
|
|
storage=monitor.storage,
|
|
file_uid=recipe['id']
|
|
)
|
|
new_recipe.save()
|
|
import_count += 1
|
|
|
|
log_entry = SyncLog(
|
|
status='SUCCESS',
|
|
msg='Imported ' + str(import_count) + ' recipes',
|
|
sync=monitor
|
|
)
|
|
log_entry.save()
|
|
|
|
monitor.last_checked = datetime.now()
|
|
monitor.save()
|
|
|
|
return True
|
|
|
|
@staticmethod
|
|
def create_share_link(recipe):
|
|
url = "https://api.dropboxapi.com/2/sharing/create_shared_link_with_settings" # noqa: E501
|
|
|
|
headers = {
|
|
"Authorization": "Bearer " + recipe.storage.token,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"path": recipe.file_uid
|
|
}
|
|
|
|
r = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
|
return r.json()
|
|
|
|
@staticmethod
|
|
def get_share_link(recipe):
|
|
url = "https://api.dropboxapi.com/2/sharing/list_shared_links"
|
|
|
|
headers = {
|
|
"Authorization": "Bearer " + recipe.storage.token,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"path": recipe.file_path,
|
|
}
|
|
|
|
r = requests.post(url, headers=headers, data=json.dumps(data))
|
|
p = r.json()
|
|
|
|
for link in p['links']:
|
|
return link['url']
|
|
|
|
response = Dropbox.create_share_link(recipe)
|
|
return response['url']
|
|
|
|
@staticmethod
|
|
def get_file(recipe):
|
|
if not recipe.link:
|
|
recipe.link = Dropbox.get_share_link(recipe)
|
|
recipe.save()
|
|
|
|
response = requests.get(
|
|
recipe.link.replace('www.dropbox.', 'dl.dropboxusercontent.')
|
|
)
|
|
|
|
return io.BytesIO(response.content)
|
|
|
|
@staticmethod
|
|
def rename_file(recipe, new_name):
|
|
url = "https://api.dropboxapi.com/2/files/move_v2"
|
|
|
|
headers = {
|
|
"Authorization": "Bearer " + recipe.storage.token,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"from_path": recipe.file_path,
|
|
"to_path": "%s/%s%s" % (
|
|
os.path.dirname(recipe.file_path),
|
|
new_name,
|
|
os.path.splitext(recipe.file_path)[1]
|
|
)
|
|
}
|
|
|
|
r = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
|
return r.json()
|
|
|
|
@staticmethod
|
|
def delete_file(recipe):
|
|
url = "https://api.dropboxapi.com/2/files/delete_v2"
|
|
|
|
headers = {
|
|
"Authorization": "Bearer " + recipe.storage.token,
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
data = {
|
|
"path": recipe.file_path
|
|
}
|
|
|
|
r = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
|
return r.json()
|