55 lines
2.9 KiB
Python
55 lines
2.9 KiB
Python
|
import re
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
from log_tools import *
|
||
|
|
from calibre_tools import CalibreTools
|
||
|
|
|
||
|
|
def validateArgs(parser, args):
|
||
|
|
if args.ip_port is None and args.library is None:
|
||
|
|
printError("Minimum one parameter required")
|
||
|
|
exit(0)
|
||
|
|
if args.ip_port is not None:
|
||
|
|
if not re.match('^\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}:\\d+$', args.ip_port) and not re.match('(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])):\\d+$', args.ip_port):
|
||
|
|
printError("Invalid IP")
|
||
|
|
exit(0)
|
||
|
|
|
||
|
|
def calibre_implementation(ip_port):
|
||
|
|
printInfo("Checking if server is accessible")
|
||
|
|
if not CalibreTools.checkServer(ip_port):
|
||
|
|
printError("Server requires authentication.")
|
||
|
|
exit(0)
|
||
|
|
printSuccess("Server is accessible")
|
||
|
|
printInfo("Getting Libraries")
|
||
|
|
libraries = []
|
||
|
|
thr = 1
|
||
|
|
if args.lib is None:
|
||
|
|
libraries = CalibreTools.getLibraries(ip_port)
|
||
|
|
if len(libraries) == 0:
|
||
|
|
printError("Could not find any libraries")
|
||
|
|
exit(0)
|
||
|
|
printSuccess(f'Found {len(libraries)} libraries')
|
||
|
|
else:
|
||
|
|
libraries.append(args.lib)
|
||
|
|
if args.thread not is None:
|
||
|
|
thr = args.thread
|
||
|
|
for library in libraries:
|
||
|
|
printInfo(f'Counting books from library: {library}')
|
||
|
|
CalibreTools.downloadBooks(ip_port, library, None, thr)
|
||
|
|
printSuccess("Download Complete!")
|
||
|
|
exit(0)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser("Calibre Dumper")
|
||
|
|
parser.add_argument('-c', '--calibre-host', type=str, dest='ip_port', action='store', help='Provide ip and port of calibre server, Format ip:port')
|
||
|
|
parser.add_argument('-l', '--library', type=str, dest='lib', action='store', help='Specify which library to download')
|
||
|
|
parser.add_argument('-t', '--threads', type=int, dest='thread', action='store', help='Specify how many download threads to use')
|
||
|
|
parser.usage = '''
|
||
|
|
CALIBRE DUMPER
|
||
|
|
1. Download all books from all libraries of calibre server using --calibre-host <IP:PORT> or -c <IP:PORT>
|
||
|
|
2. Specify library using --library <LIBRARY NAME> or -l <LIBRARY NAME>
|
||
|
|
3. Specify how many download threads to use --threads <NUMBER> or -t <NUMBER>
|
||
|
|
'''
|
||
|
|
args = parser.parse_args()
|
||
|
|
validateArgs(parser, args)
|
||
|
|
if args.ip_port is not None:
|
||
|
|
calibre_implementation(args.ip_port)
|