diff --git a/README.md b/README.md index 4f68950..d0ff15f 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,3 @@ # Cloudy +easy tool to zip and upload directories to own/nextcloud \ No newline at end of file diff --git a/cloudy.py b/cloudy.py new file mode 100644 index 0000000..8b7d546 --- /dev/null +++ b/cloudy.py @@ -0,0 +1,155 @@ +__author__ = "Lukas Mahler" +__version__ = "0.0.1" +__date__ = "07.11.2021" +__email__ = "lm@ankerlab.de" +__status__ = "Development" + + +try: + # Defaults + import os + import sys + import stat + import shutil + import requests + import tempfile + import datetime + + # Customs + import pyzipper + import owncloud + + # Self + from src import util + +except ImportError as e: + print(f"ERROR: Missing Module [{e}]") + sys.exit() + + +def del_rw(action, name, exc): + os.chmod(name, stat.S_IWRITE) + os.remove(name) + + +def create_duplicate_dir(folder): + """ + Creates a duplicate of the folder in the tempdir. + Throws errors when there is a Thumbs.db blocking... + """ + temp_dir = tempfile.gettempdir() + date = datetime.datetime.now() + ext = date.strftime("%Y-%m-%d") + src = folder + dst = os.path.join(temp_dir, ext + "_" + "Dump") + + print("[*] Trying to create a Duplicate, please wait...") + if os.path.exists(dst): + try: + shutil.rmtree(dst, onerror=del_rw) + except PermissionError as e: + print(f"[X] Error: {e}") + exit(1) + + shutil.copytree(src, dst) + print("[*] Successfully created Duplicate") + + return dst + + +# inspired by https://stackoverflow.com/questions/60087965/how-to-zip-a-folder-in-python-with-password +def zip_folder(folder): + """ + Zips the duplicated folder in the tempdir. + """ + + print("[*] Trying to create corresponding zipfile") + + zipfilename = folder + ".zip" + parent_folder = os.path.dirname(folder) + contents = os.walk(folder) + + with pyzipper.AESZipFile(zipfilename, 'w', compression=pyzipper.ZIP_DEFLATED, encryption=pyzipper.WZ_AES) as zf: + + if config["Other"]["zippw"] != "": + zf.pwd = bytes(config["Other"]["zippw"], encoding="utf8") + + for root, folders, files in contents: + # Include all subfolders, including empty ones. + for folder_name in folders: + try: + absolute_path = os.path.join(root, folder_name) + relative_path = absolute_path.replace(parent_folder + '\\', '') + print(f" [+] {absolute_path}") + zf.write(absolute_path, relative_path) + except: + print(f" [-] couldn't add {absolute_path}") + continue + for file_name in files: + try: + absolute_path = os.path.join(root, file_name) + relative_path = absolute_path.replace(parent_folder + '\\', '') + print(f" [+] {absolute_path}") + zf.write(absolute_path, relative_path) + except: + print(f" [-] couldn't add {absolute_path}") + continue + + print("[*] Successfully created corresponding zipfile") + + return zipfilename + + +def upload_to_nextcloud(file): + """ + Uploads the file to Own/Nextcloud + """ + + url = config["Connection"]["host"] + usr = config["Auth"]["user"] + pwd = config["Auth"]["password"] + pth = config["Other"]["savepath"] + + # Test Connection + r = requests.head(url) + httpc = str(r.status_code)[0] + if httpc == "4" or httpc == "5": + print(f"ERROR: HTTP Statuscode for URL [{url}] is [{r.status_code}]") + return + + try: + nxt = owncloud.Client(url) + nxt.login(usr, pwd) + except Exception as e: + print("ERROR: ", e) + return + + print("[*] Starting the file Upload...") + nxt.makedirs(pth, exist_ok=True) + nxt.put_file(pth, file) + + print(f"[*] Finished uploading to {url}") + + +def clean_temp(tempdir): + if os.path.exists(tempdir): + print("Cleaning up the tempdir [", tempdir, "]...") + shutil.rmtree(tempdir, ignore_errors=True) + + +def main(): + + # This will be uploaded + folder = os.path.join(os.environ['USERPROFILE'], 'Desktop', "Dump") + + # Load toml config + global config + config = util.getConf("prod.toml") + + dupe = create_duplicate_dir(folder) + zipped = zip_folder(dupe) + upload_to_nextcloud(zipped) + + +if __name__ == "__main__": + main() diff --git a/src/template.toml b/src/template.toml new file mode 100644 index 0000000..281c489 --- /dev/null +++ b/src/template.toml @@ -0,0 +1,10 @@ +[Connection] +host = "myexamplecloud.com" + +[Auth] +user = "exampleuser" +password = "examplepassword" + +[Other] +savepath = "Zips/" +zippw = "" \ No newline at end of file diff --git a/src/util.py b/src/util.py new file mode 100644 index 0000000..5db09ea --- /dev/null +++ b/src/util.py @@ -0,0 +1,53 @@ +""" +TBD +""" + +__author__ = "Lukas Mahler" +__version__ = "0.0.0" +__date__ = "13.10.2021" +__email__ = "m@hler.eu" +__status__ = "Development" + + +# Default +import shutil +import os.path + + +# Custom +import toml + + +def getConf(fname): + """ + """ + if fname.endswith(".toml"): + if os.path.isfile(fname): + try: + config = toml.load(fname) + checkConf(config) + return config + except ValueError as e: + print(f"The provided '.toml' is probably invalid, returned error:\n{e}") + exit(1) + else: + print(f"Couldn't locate the '.toml' file [{fname}].") + print("Creating a new '.toml' file from template, please edit and restart.") + shutil.copy("src/template.toml", fname) + exit(1) + else: + print(f"The provided config file [{fname}] is not a '.toml' file.") + print("Creating a new '.toml' file from template, please edit and restart.") + shutil.copy("src/template.toml", "prod.toml") + exit(1) + + +def checkConf(config): + """ + TODO check if keys exist + """ + pass + + +if __name__ == "__main__": + exit()