mirror of
https://github.com/noDRM/DeDRM_tools.git
synced 2024-11-17 03:16:09 +06:00
278 lines
9.2 KiB
Python
278 lines
9.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# CLI interface for the DeDRM plugin (useable without Calibre, too)
|
|
|
|
from __future__ import absolute_import, print_function
|
|
|
|
# Copyright © 2021 NoDRM
|
|
|
|
OPT_SHORT_TO_LONG = [
|
|
["c", "config"],
|
|
["e", "extract"],
|
|
["f", "force"],
|
|
["h", "help"],
|
|
["i", "import"],
|
|
["o", "output"],
|
|
["p", "password"],
|
|
["q", "quiet"],
|
|
["t", "test"],
|
|
["u", "username"],
|
|
["v", "verbose"],
|
|
]
|
|
|
|
#@@CALIBRE_COMPAT_CODE@@
|
|
|
|
import os, sys
|
|
|
|
|
|
global _additional_data
|
|
global _additional_params
|
|
global _function
|
|
_additional_data = []
|
|
_additional_params = []
|
|
_function = None
|
|
|
|
global config_file_path
|
|
config_file_path = "dedrm.json"
|
|
|
|
def print_fname(f, info):
|
|
print(" " + f.ljust(15) + " " + info)
|
|
|
|
def print_opt(short, long, info):
|
|
if short is None:
|
|
short = " "
|
|
else:
|
|
short = " -" + short
|
|
|
|
if long is None:
|
|
long = " "
|
|
else:
|
|
long = "--" + long.ljust(16)
|
|
|
|
print(short + " " + long + " " + info, file=sys.stderr)
|
|
|
|
def print_std_usage(name, param_string):
|
|
print("Usage: ", file=sys.stderr)
|
|
if "calibre" in sys.modules:
|
|
print(" calibre-debug -r \"DeDRM\" -- "+name+" " + param_string, file=sys.stderr)
|
|
else:
|
|
print(" python3 DeDRM_plugin.zip "+name+" "+param_string, file=sys.stderr)
|
|
|
|
def print_err_header():
|
|
from __init__ import PLUGIN_NAME, PLUGIN_VERSION # type: ignore
|
|
|
|
print(PLUGIN_NAME + " v" + PLUGIN_VERSION + " - DRM removal plugin by noDRM")
|
|
print()
|
|
|
|
def print_help():
|
|
from __version import PLUGIN_NAME, PLUGIN_VERSION
|
|
print(PLUGIN_NAME + " v" + PLUGIN_VERSION + " - DRM removal plugin by noDRM")
|
|
print("Based on DeDRM Calibre plugin by Apprentice Harper, Apprentice Alf and others.")
|
|
print("See https://github.com/noDRM/DeDRM_tools for more information.")
|
|
print()
|
|
if "calibre" in sys.modules:
|
|
print("This plugin can be run through Calibre - like you are doing right now - ")
|
|
print("but it can also be executed with a standalone Python interpreter.")
|
|
else:
|
|
print("This plugin can either be imported into Calibre, or be executed directly")
|
|
print("through Python like you are doing right now.")
|
|
print()
|
|
print("Available functions:")
|
|
print_fname("passhash", "Manage Adobe PassHashes")
|
|
print_fname("remove_drm", "Remove DRM from one or multiple books")
|
|
print()
|
|
|
|
# TODO: All parameters that are global should be listed here.
|
|
|
|
def print_credits():
|
|
from __version import PLUGIN_NAME, PLUGIN_VERSION
|
|
print(PLUGIN_NAME + " v" + PLUGIN_VERSION + " - Calibre DRM removal plugin by noDRM")
|
|
print("Based on DeDRM Calibre plugin by Apprentice Harper, Apprentice Alf and others.")
|
|
print("See https://github.com/noDRM/DeDRM_tools for more information.")
|
|
print()
|
|
print("Credits:")
|
|
print(" - noDRM for the current release of the DeDRM plugin")
|
|
print(" - Apprentice Alf and Apprentice Harper for the previous versions of the DeDRM plugin")
|
|
print(" - The Dark Reverser for the Mobipocket and eReader script")
|
|
print(" - i ♥ cabbages for the Adobe Digital Editions scripts")
|
|
print(" - Skindle aka Bart Simpson for the Amazon Kindle for PC script")
|
|
print(" - CMBDTC for Amazon Topaz DRM removal script")
|
|
print(" - some_updates, clarknova and Bart Simpson for Amazon Topaz conversion scripts")
|
|
print(" - DiapDealer for the first calibre plugin versions of the tools")
|
|
print(" - some_updates, DiapDealer, Apprentice Alf and mdlnx for Amazon Kindle/Mobipocket tools")
|
|
print(" - some_updates for the DeDRM all-in-one Python tool")
|
|
print(" - Apprentice Alf for the DeDRM all-in-one AppleScript tool")
|
|
|
|
|
|
def handle_single_argument(arg, next):
|
|
used_up = 0
|
|
global _additional_params
|
|
global config_file_path
|
|
|
|
if arg in ["--username", "--password", "--output", "--outputdir"]:
|
|
used_up = 1
|
|
_additional_params.append(arg)
|
|
if next is None or len(next) == 0:
|
|
print_err_header()
|
|
print("Missing parameter for argument " + arg, file=sys.stderr)
|
|
sys.exit(1)
|
|
else:
|
|
_additional_params.append(next[0])
|
|
|
|
elif arg == "--config":
|
|
if next is None or len(next) == 0:
|
|
print_err_header()
|
|
print("Missing parameter for argument " + arg, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
config_file_path = next[0]
|
|
used_up = 1
|
|
|
|
elif arg in ["--help", "--credits", "--verbose", "--quiet", "--extract", "--import", "--overwrite", "--force"]:
|
|
_additional_params.append(arg)
|
|
|
|
|
|
else:
|
|
print_err_header()
|
|
print("Unknown argument: " + arg, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
# Used up 0 additional arguments
|
|
return used_up
|
|
|
|
|
|
|
|
def handle_data(data):
|
|
global _function
|
|
global _additional_data
|
|
|
|
if _function is None:
|
|
_function = str(data)
|
|
else:
|
|
_additional_data.append(str(data))
|
|
|
|
def execute_action(action, filenames, params):
|
|
print("Executing '{0}' on file(s) {1} with parameters {2}".format(action, str(filenames), str(params)), file=sys.stderr)
|
|
|
|
if action == "help":
|
|
print_help()
|
|
sys.exit(0)
|
|
|
|
elif action == "passhash":
|
|
from standalone.passhash import perform_action
|
|
perform_action(params, filenames)
|
|
|
|
elif action == "remove_drm":
|
|
if not os.path.isfile(os.path.abspath(config_file_path)):
|
|
print("Config file missing ...")
|
|
|
|
from standalone.remove_drm import perform_action
|
|
perform_action(params, filenames)
|
|
|
|
elif action == "config":
|
|
import prefs
|
|
config = prefs.DeDRM_Prefs(os.path.abspath(config_file_path))
|
|
print(config["adeptkeys"])
|
|
|
|
else:
|
|
print("Command '"+action+"' is unknown.", file=sys.stderr)
|
|
|
|
|
|
def main(argv):
|
|
arguments = argv
|
|
skip_opts = False
|
|
|
|
# First element is always the ZIP name, remove that.
|
|
if not arguments[0].lower().endswith(".zip") and not "calibre" in sys.modules:
|
|
print("Warning: File name does not end in .zip ...")
|
|
print(arguments)
|
|
arguments.pop(0)
|
|
|
|
while len(arguments) > 0:
|
|
arg = arguments.pop(0)
|
|
|
|
if arg == "--":
|
|
skip_opts = True
|
|
continue
|
|
|
|
if not skip_opts:
|
|
if arg.startswith("--"):
|
|
# Give the current arg, plus all remaining ones.
|
|
# Return the number of additional args we used.
|
|
used = handle_single_argument(arg, arguments)
|
|
for _ in range(used):
|
|
# Function returns number of additional arguments that were
|
|
# "used up" by that argument.
|
|
# Remove that amount of arguments from the list.
|
|
try:
|
|
arguments.pop(0)
|
|
except:
|
|
pass
|
|
continue
|
|
elif arg.startswith("-"):
|
|
single_args = list(arg[1:])
|
|
# single_args is now a list of single chars, for when you call the program like "ls -alR"
|
|
# with multiple single-letter options combined.
|
|
while len(single_args) > 0:
|
|
c = single_args.pop(0)
|
|
|
|
# See if we have a long name for that option.
|
|
for wrapper in OPT_SHORT_TO_LONG:
|
|
if wrapper[0] == c:
|
|
c = "--" + wrapper[1]
|
|
break
|
|
else:
|
|
c = "-" + c
|
|
# c is now the long term (unless there is no long version, then it's the short version).
|
|
|
|
if len(single_args) > 0:
|
|
# If we have more short arguments, the argument for this one must be None.
|
|
handle_single_argument(c, None)
|
|
used = 0
|
|
else:
|
|
# If not, then there might be parameters for this short argument.
|
|
used = handle_single_argument(c, arguments)
|
|
|
|
for _ in range(used):
|
|
# Function returns number of additional arguments that were
|
|
# "used up" by that argument.
|
|
# Remove that amount of arguments from the list.
|
|
try:
|
|
arguments.pop(0)
|
|
except:
|
|
pass
|
|
|
|
continue
|
|
|
|
handle_data(arg)
|
|
|
|
|
|
if _function is None and "--credits" in _additional_params:
|
|
print_credits()
|
|
sys.exit(0)
|
|
|
|
if _function is None and "--help" in _additional_params:
|
|
print_help()
|
|
sys.exit(0)
|
|
|
|
if _function is None:
|
|
print_help()
|
|
sys.exit(1)
|
|
|
|
# Okay, now actually begin doing stuff.
|
|
# This function gets told what to do and gets additional data (filenames).
|
|
# It also receives additional parameters.
|
|
# The rest of the code will be in different Python files.
|
|
execute_action(_function.lower(), _additional_data, _additional_params)
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# NOTE: This MUST not do anything else other than calling main()
|
|
# All the code must be in main(), not in here.
|
|
import sys
|
|
main(sys.argv) |