forked from l3uddz/plex_autoscan
-
Notifications
You must be signed in to change notification settings - Fork 5
/
rclone.py
44 lines (38 loc) · 2.38 KB
/
rclone.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import subprocess
import os
import logging
logger = logging.getLogger("RCLONE")
class RcloneDecoder:
def __init__(self, binary, crypt_mappings, config):
self._binary = binary
if self._binary == "" or not os.path.isfile(binary):
self._binary = os.path.normpath(subprocess.check_output(["which", "rclone"]).decode().rstrip('\n'))
logger.debug("Rclone binary path located as: '%s'", binary)
self._config = config
self._crypt_mappings = crypt_mappings
def decode_path(self, path):
for crypt_dir, mapped_remotes in self._crypt_mappings.items():
# Isolate root/file path and attempt to locate entry in mappings
file_path = path.replace(crypt_dir,'')
logger.debug("Encoded file path identified as: '%s'", file_path)
if path.lower().startswith(crypt_dir.lower()):
for mapped_remote in mapped_remotes:
logger.debug("Crypt base directory identified as: '%s'", crypt_dir)
logger.debug("Crypt base directory '%s' has mapping defined in config as remote '%s'.", crypt_dir, mapped_remote)
logger.info("Attempting to decode...")
logger.debug("Raw query is: '%s'", " ".join([self._binary, "--config", self._config, "cryptdecode", mapped_remote, file_path]))
try:
decoded = subprocess.check_output([self._binary, "--config", self._config, "cryptdecode", mapped_remote, file_path], stderr=subprocess.STDOUT).decode('utf-8').rstrip('\n')
except subprocess.CalledProcessError as e:
logger.error("Command '%s' returned with error (code %s): %s", e.cmd, e.returncode, e.output)
return None
decoded = decoded.split(' ',1)[1].lstrip()
if "failed" in decoded.lower():
logger.error("Failed to decode path: '%s'", file_path)
else:
logger.debug("Decoded path of '%s' is: '%s'", file_path, os.path.join(crypt_dir, decoded))
logger.info("Decode successful.")
return [os.path.join(crypt_dir, decoded)]
else:
logger.debug("Ignoring crypt decode for path '%s' because '%s' was not matched from 'CRYPT_MAPPINGS'.", path, crypt_dir)
return None