31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
import os
|
|
import requests
|
|
|
|
|
|
def backup_book(api, env, selected_book, backup_extension, backup_extension_fileType):
|
|
base_url = env['base_url']
|
|
backup_dir = env['backup_dir']
|
|
headers = {
|
|
"Authorization": f"Token {env['token_id']}:{env['token_secret']}"
|
|
}
|
|
|
|
pages_list = api.get_pages_list()['data']
|
|
chapter_list = api.get_chapters_list()['data']
|
|
|
|
print(f"Backup type: {backup_extension}")
|
|
|
|
for chapter in chapter_list:
|
|
for page in pages_list:
|
|
if page['book_id'] == selected_book['id']:
|
|
if chapter['id'] == page['chapter_id']:
|
|
folder = os.path.join(backup_dir, chapter['name'])
|
|
filename = os.path.join(folder, page['name'] + backup_extension_fileType)
|
|
print(f"Backing up: {filename}")
|
|
if not os.path.exists(folder):
|
|
os.makedirs(folder, exist_ok=True)
|
|
response = requests.get(
|
|
f'{base_url}/api/pages/{page["id"]}/export/{backup_extension}',
|
|
headers=headers
|
|
)
|
|
with open(filename, 'wb') as file:
|
|
file.write(response.content) |