From b969aa886e6d83ce815e36381f0b7d60b3d49057 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:34:45 +0200 Subject: [PATCH 01/21] ignore /.vscode --- .gitignore | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index 90c0433..0800ae8 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,4 @@ __pycache__ /tmp -/.idea +/.vscode /dist From 4ab0a575b7b9439c219e60d3768424d2dc6af40a Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:35:12 +0200 Subject: [PATCH 02/21] pyproject/ruff --- pyproject.toml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index ac386d1..2c1938b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,3 +30,6 @@ requires-python = '>= 3.10' [tool.hatch.version] path = 'rwx/__init__.py' + +[tool.ruff] +select = ["ALL"] From 566617758228ea575e0f518f0388e7ae15ec9d16 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:37:23 +0200 Subject: [PATCH 03/21] pyproject/" --- pyproject.toml | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 2c1938b..ccd0aa1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,35 +1,35 @@ [build-system] -requires = ['hatchling'] -build-backend = 'hatchling.build' +requires = ["hatchling"] +build-backend = "hatchling.build" [project] authors = [ - { name = 'Marc Beninca', email = 'git@marc.beninca.link' }, + { name = "Marc Beninca", email = "git@marc.beninca.link" }, ] maintainers = [ - { name = 'Marc Beninca', email = 'git@marc.beninca.link' }, + { name = "Marc Beninca", email = "git@marc.beninca.link" }, ] classifiers = [ - 'Programming Language :: Python :: 3', - 'License :: OSI Approved :: GNU Affero General Public License v3', - 'Operating System :: OS Independent', + "Programming Language :: Python :: 3", + "License :: OSI Approved :: GNU Affero General Public License v3", + "Operating System :: OS Independent", ] dependencies = [] -description = 'Read Write eXecute' -dynamic = ['version'] +description = "Read Write eXecute" +dynamic = ["version"] keywords = [] -license-files = { paths = ['license.md'] } -name = 'rwx' -readme = 'readme.md' -requires-python = '>= 3.10' +license-files = { paths = ["license.md"] } +name = "rwx" +readme = "readme.md" +requires-python = ">= 3.10" [project.scripts] -# command = 'package.module:function' +# command = "package.module:function" [project.urls] [tool.hatch.version] -path = 'rwx/__init__.py' +path = "rwx/__init__.py" [tool.ruff] select = ["ALL"] From b197a156896d0e8cae8869122e05a1f4d5434027 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:38:08 +0200 Subject: [PATCH 04/21] linting --- rwx/prj/sphinx.py | 2 +- rwx/txt/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rwx/prj/sphinx.py b/rwx/prj/sphinx.py index 4da1ebf..392e7c2 100644 --- a/rwx/prj/sphinx.py +++ b/rwx/prj/sphinx.py @@ -10,7 +10,7 @@ class SphinxProject(Project): super().__init__(file_path) def build(self): - output_root: str = path.join(self.root, 'out') + output_root: str = path.join(self.root, "out") wipe(output_root) arguments: list[str] = [ "-E", diff --git a/rwx/txt/__init__.py b/rwx/txt/__init__.py index edb0624..369202c 100644 --- a/rwx/txt/__init__.py +++ b/rwx/txt/__init__.py @@ -1 +1 @@ -CHARSET = 'UTF-8' +CHARSET = "UTF-8" From fdbd850d231104fc6264536e2ef7d082152adf9e Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:41:34 +0200 Subject: [PATCH 05/21] lint rwx --- rwx/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rwx/__init__.py b/rwx/__init__.py index b8023d8..a721b9a 100644 --- a/rwx/__init__.py +++ b/rwx/__init__.py @@ -1 +1,2 @@ -__version__ = '0.0.1' +"""Read Write eXecute.""" +__version__ = "0.0.1" From a4f49d6ac09549d958588b3eec84cedcaaa0c412 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:43:22 +0200 Subject: [PATCH 06/21] exception --- rwx/err/__init__.py | 2 ++ 1 file changed, 2 insertions(+) create mode 100644 rwx/err/__init__.py diff --git a/rwx/err/__init__.py b/rwx/err/__init__.py new file mode 100644 index 0000000..b996913 --- /dev/null +++ b/rwx/err/__init__.py @@ -0,0 +1,2 @@ +class Exception(Exception): + pass From c7db231dc2a5e6244e06dea770a4cb3f6a6ce430 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:50:13 +0200 Subject: [PATCH 07/21] lint grub --- rwx/grub/__init__.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/rwx/grub/__init__.py b/rwx/grub/__init__.py index b931ee3..a11cf7d 100644 --- a/rwx/grub/__init__.py +++ b/rwx/grub/__init__.py @@ -1,17 +1,17 @@ import cmd import ps -cmd.need('grub-mkimage') +cmd.need("grub-mkimage") -COMPRESSION = 'xz' +COMPRESSION = "xz" ENV_BYTES = 1024 -ENV_COMMENT = '#' -ENV_HEADER = f'''{ENV_COMMENT} GRUB Environment Block -''' +ENV_COMMENT = "#" +ENV_HEADER = f"""{ENV_COMMENT} GRUB Environment Block +""" MODULES = { - 'i386-pc': [ - ('biosdisk',), - ('ntldr',), + "i386-pc": [ + ("biosdisk",), + ("ntldr",), ] } @@ -19,14 +19,14 @@ MODULES = { def make_image(image_format: str, image_path: str, modules: list[str], memdisk_path: str, pubkey_path: str = None) -> None: args = [ - ('grub-mkimage',), - ('--compress', COMPRESSION), - ('--format', image_format), - ('--output', image_path), - ('--memdisk', memdisk_path), + ("grub-mkimage",), + ("--compress", COMPRESSION), + ("--format", image_format), + ("--output", image_path), + ("--memdisk", memdisk_path), ] if pubkey_path: - args.append(('--pubkey', pubkey_path)) + args.append(("--pubkey", pubkey_path)) args.extend(modules) if modules := MODULES.get(image_format, None): args.extend(modules) From 592023d2140ea11b5b854072a6893bbe8d8973f6 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:51:59 +0200 Subject: [PATCH 08/21] lint fs --- rwx/fs/__init__.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/rwx/fs/__init__.py b/rwx/fs/__init__.py index ecaebf0..2a6af2c 100644 --- a/rwx/fs/__init__.py +++ b/rwx/fs/__init__.py @@ -3,13 +3,13 @@ import shutil from rwx import ps -CHARSET = 'UTF-8' +CHARSET = "UTF-8" def create_image(file_path: str, size_bytes: int): ps.run( - ('qemu-img', 'create'), - ('-f', 'qcow2'), + ("qemu-img", "create"), + ("-f", "qcow2"), (file_path, size_bytes), ) @@ -20,17 +20,17 @@ def empty_file(path: str): def get_mount_uuid(path: str): return ps.run_line( - ('findmnt',), - ('--noheadings',), - ('--output', 'UUID'), + ("findmnt",), + ("--noheadings",), + ("--output", "UUID"), (path,), ) def get_path_mount(path: str): return ps.run_line( - ('stat',), - ('--format', '%m'), + ("stat",), + ("--format", "%m"), (path,), ) @@ -44,7 +44,7 @@ def make_directory(path: str): def read_file(file_path: str): - with open(file_path, 'br') as file_object: + with open(file_path, "br") as file_object: return file_object.read() @@ -66,5 +66,5 @@ def wipe(path: str): def write(file_path: str, text: str, charset=CHARSET): - with open(file_path, 'bw') as file_object: + with open(file_path, "bw") as file_object: file_object.write(text.encode(charset)) From 470d7aa1e92480ff9e9fa8ccfeb6a2bbd5b4b2e3 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:52:56 +0200 Subject: [PATCH 09/21] lint deb --- rwx/deb/__init__.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/rwx/deb/__init__.py b/rwx/deb/__init__.py index a773123..7059e93 100644 --- a/rwx/deb/__init__.py +++ b/rwx/deb/__init__.py @@ -1,17 +1,17 @@ import cmd import ps -cmd.need('debootstrap') +cmd.need("debootstrap") -BOOTSTRAP_ARCHITECTURE = 'amd64' -BOOTSTRAP_VARIANT = 'minbase' +BOOTSTRAP_ARCHITECTURE = "amd64" +BOOTSTRAP_VARIANT = "minbase" def bootstrap(root_path: str, suite: str, mirror_location: str): command = [ - ('debootstrap',), - ('--arch', BOOTSTRAP_ARCHITECTURE), - ('--variant', BOOTSTRAP_VARIANT), + ("debootstrap",), + ("--arch", BOOTSTRAP_ARCHITECTURE), + ("--variant", BOOTSTRAP_VARIANT), (suite,), (root_path,), (mirror_location,), From 67b8d4c9819ad3f1f6f2166b4d757f03e656e16e Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:53:51 +0200 Subject: [PATCH 10/21] lint squashfs --- rwx/cmd/squashfs/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rwx/cmd/squashfs/__init__.py b/rwx/cmd/squashfs/__init__.py index 637890a..4f793ca 100644 --- a/rwx/cmd/squashfs/__init__.py +++ b/rwx/cmd/squashfs/__init__.py @@ -1,14 +1,14 @@ import ps import rwx.cmd -rwx.cmd.need('mksquashfs') +rwx.cmd.need("mksquashfs") def mksquashfs(input_root: str, output_file: str): ps.run([ - 'mksquashfs', + "mksquashfs", input_root, output_file, - '-comp', 'zstd', - '-Xcompression-level', str(18), + "-comp", "zstd", + "-Xcompression-level", str(18), ]) From bf5a8a65d781c416aba5f6684150421bf8176893 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:55:13 +0200 Subject: [PATCH 11/21] lint cmd --- rwx/cmd/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rwx/cmd/__init__.py b/rwx/cmd/__init__.py index 9403d24..6ecb47c 100644 --- a/rwx/cmd/__init__.py +++ b/rwx/cmd/__init__.py @@ -4,10 +4,10 @@ packages: list[str] = [] def need(command: str) -> None: match command: - case 'debootstrap': - package = 'debootstrap' - case 'mksquashfs' | 'unsquashfs': - package = 'squashfs-tools' + case "debootstrap": + package = "debootstrap" + case "mksquashfs" | "unsquashfs": + package = "squashfs-tools" case _: package = None if package: From 517da87ffb51c2577a388a42c9627f8775516317 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 09:56:59 +0200 Subject: [PATCH 12/21] lint main --- rwx/__main__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/rwx/__main__.py b/rwx/__main__.py index 0f73041..0f15b2e 100755 --- a/rwx/__main__.py +++ b/rwx/__main__.py @@ -5,14 +5,14 @@ import os import fs -if __name__ == '__main__': +if __name__ == "__main__": file_path = os.path.realpath(__file__) root_path = os.path.dirname(file_path) - directory_path = os.path.join(root_path, 'tmp') - file_path = os.path.join(directory_path, 'file') + directory_path = os.path.join(root_path, "tmp") + file_path = os.path.join(directory_path, "file") fs.wipe(directory_path) fs.make_directory(directory_path) - fs.write(file_path, 'Martine écrit beaucoup.') + fs.write(file_path, "Martine écrit beaucoup.") fs.empty_file(file_path) - fs.write(file_path, 'Martine écrit moins.') + fs.write(file_path, "Martine écrit moins.") From e84417c077d0bdd489a93c4703bf5bd3738e09a1 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 10:01:24 +0200 Subject: [PATCH 13/21] lint fs --- rwx/fs/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rwx/fs/__init__.py b/rwx/fs/__init__.py index 2a6af2c..e769358 100644 --- a/rwx/fs/__init__.py +++ b/rwx/fs/__init__.py @@ -15,7 +15,7 @@ def create_image(file_path: str, size_bytes: int): def empty_file(path: str): - write(path, str()) + write(path, "") def get_mount_uuid(path: str): From fc3e191dc1aacd97f974836b8584cccae8572153 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 10:04:54 +0200 Subject: [PATCH 14/21] lint prj --- rwx/prj/__init__.py | 2 +- rwx/prj/sphinx.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rwx/prj/__init__.py b/rwx/prj/__init__.py index f0a2fe5..227a686 100644 --- a/rwx/prj/__init__.py +++ b/rwx/prj/__init__.py @@ -2,7 +2,7 @@ from os import path class Project: - def __init__(self, file_path: str): + def __init__(self, file_path: str) -> None: self.file: str = path.realpath(file_path) self.root: str = path.dirname(self.file) self.name: str = path.basename(self.root) diff --git a/rwx/prj/sphinx.py b/rwx/prj/sphinx.py index 392e7c2..a71bc35 100644 --- a/rwx/prj/sphinx.py +++ b/rwx/prj/sphinx.py @@ -6,7 +6,7 @@ from rwx.prj import Project class SphinxProject(Project): - def __init__(self, file_path: str): + def __init__(self, file_path: str) -> None: super().__init__(file_path) def build(self): From b44b6bd4d4e2c39d545e6314b5019c59755fcaa0 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 10:06:25 +0200 Subject: [PATCH 15/21] lint deb --- rwx/deb/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/rwx/deb/__init__.py b/rwx/deb/__init__.py index 7059e93..bfb8a2f 100644 --- a/rwx/deb/__init__.py +++ b/rwx/deb/__init__.py @@ -16,5 +16,4 @@ def bootstrap(root_path: str, suite: str, mirror_location: str): (root_path,), (mirror_location,), ] - completed_process = ps.run(command) - return completed_process + return ps.run(command) From c7022836966379a640d007c2d8db8fd03d9f7d73 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 12:49:49 +0200 Subject: [PATCH 16/21] lint commas --- rwx/grub/__init__.py | 2 +- rwx/log/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rwx/grub/__init__.py b/rwx/grub/__init__.py index a11cf7d..3a70ad4 100644 --- a/rwx/grub/__init__.py +++ b/rwx/grub/__init__.py @@ -12,7 +12,7 @@ MODULES = { "i386-pc": [ ("biosdisk",), ("ntldr",), - ] + ], } diff --git a/rwx/log/__init__.py b/rwx/log/__init__.py index 2cf0b09..4d28f82 100644 --- a/rwx/log/__init__.py +++ b/rwx/log/__init__.py @@ -4,7 +4,7 @@ import sys def get_logger(name: str) -> logging.Logger: formatter = logging.Formatter( - "%(name)s: %(asctime)s | %(levelname)s | %(filename)s:%(lineno)s | %(process)d >>> %(message)s" + "%(name)s: %(asctime)s | %(levelname)s | %(filename)s:%(lineno)s | %(process)d >>> %(message)s", ) # file_handler = logging.FileHandler('log.txt') From aad5c140ddec8c365e3cd3f86a8cff14e50fb758 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 12:52:19 +0200 Subject: [PATCH 17/21] =?UTF-8?q?=E2=88=92#?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- rwx/log/__init__.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/rwx/log/__init__.py b/rwx/log/__init__.py index 4d28f82..b951ac9 100644 --- a/rwx/log/__init__.py +++ b/rwx/log/__init__.py @@ -1,22 +1,16 @@ import logging import sys - - def get_logger(name: str) -> logging.Logger: formatter = logging.Formatter( "%(name)s: %(asctime)s | %(levelname)s | %(filename)s:%(lineno)s | %(process)d >>> %(message)s", ) - - # file_handler = logging.FileHandler('log.txt') - # file_handler.setFormatter(formatter) - # file_handler.setLevel(logging.INFO) - + # out_handler = logging.StreamHandler(stream=sys.stdout) out_handler.setFormatter(formatter) out_handler.setLevel(logging.INFO) - + # logger = logging.getLogger(name) - # logger.addHandler(file_handler) logger.addHandler(out_handler) logger.setLevel(logging.INFO) + # return logger From 9349775193ac5f90364ce8287eb95748cd75fc19 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 15:23:05 +0200 Subject: [PATCH 18/21] log/stream --- rwx/log/__init__.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/rwx/log/__init__.py b/rwx/log/__init__.py index b951ac9..a567492 100644 --- a/rwx/log/__init__.py +++ b/rwx/log/__init__.py @@ -1,6 +1,6 @@ import logging import sys -def get_logger(name: str) -> logging.Logger: +def get_file_logger(name: str) -> logging.Logger: formatter = logging.Formatter( "%(name)s: %(asctime)s | %(levelname)s | %(filename)s:%(lineno)s | %(process)d >>> %(message)s", ) @@ -14,3 +14,13 @@ def get_logger(name: str) -> logging.Logger: logger.setLevel(logging.INFO) # return logger +def get_stream_logger() -> logging.Logger: + out_handler = logging.StreamHandler(stream=sys.stdout) + out_handler.setLevel(logging.INFO) + # + logger = logging.getLogger() + logger.addHandler(out_handler) + logger.setLevel(logging.INFO) + # + return logger +stream = get_stream_logger() From adbdf47ef49c5963cea2c4a38fe1c26e603205b7 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 15:27:33 +0200 Subject: [PATCH 19/21] lint |None --- rwx/grub/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rwx/grub/__init__.py b/rwx/grub/__init__.py index 3a70ad4..7690a49 100644 --- a/rwx/grub/__init__.py +++ b/rwx/grub/__init__.py @@ -17,7 +17,7 @@ MODULES = { def make_image(image_format: str, image_path: str, modules: list[str], - memdisk_path: str, pubkey_path: str = None) -> None: + memdisk_path: str, pubkey_path: str | None = None) -> None: args = [ ("grub-mkimage",), ("--compress", COMPRESSION), From 88434db29aa09c7175cece250bb39d4df75e4035 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 15:31:15 +0200 Subject: [PATCH 20/21] lint fix --- rwx/__main__.py | 1 - rwx/cmd/__init__.py | 5 ++--- rwx/cmd/squashfs/__init__.py | 1 + rwx/deb/__init__.py | 1 + rwx/grub/__init__.py | 1 + rwx/log/__init__.py | 2 ++ rwx/prj/sphinx.py | 1 + 7 files changed, 8 insertions(+), 4 deletions(-) diff --git a/rwx/__main__.py b/rwx/__main__.py index 0f15b2e..a857803 100755 --- a/rwx/__main__.py +++ b/rwx/__main__.py @@ -4,7 +4,6 @@ import os import fs - if __name__ == "__main__": file_path = os.path.realpath(__file__) root_path = os.path.dirname(file_path) diff --git a/rwx/cmd/__init__.py b/rwx/cmd/__init__.py index 6ecb47c..de764fa 100644 --- a/rwx/cmd/__init__.py +++ b/rwx/cmd/__init__.py @@ -10,6 +10,5 @@ def need(command: str) -> None: package = "squashfs-tools" case _: package = None - if package: - if package not in packages: - packages.append(package) + if package and package not in packages: + packages.append(package) diff --git a/rwx/cmd/squashfs/__init__.py b/rwx/cmd/squashfs/__init__.py index 4f793ca..343461e 100644 --- a/rwx/cmd/squashfs/__init__.py +++ b/rwx/cmd/squashfs/__init__.py @@ -1,4 +1,5 @@ import ps + import rwx.cmd rwx.cmd.need("mksquashfs") diff --git a/rwx/deb/__init__.py b/rwx/deb/__init__.py index bfb8a2f..e9c6a9e 100644 --- a/rwx/deb/__init__.py +++ b/rwx/deb/__init__.py @@ -1,4 +1,5 @@ import cmd + import ps cmd.need("debootstrap") diff --git a/rwx/grub/__init__.py b/rwx/grub/__init__.py index 7690a49..57e313c 100644 --- a/rwx/grub/__init__.py +++ b/rwx/grub/__init__.py @@ -1,4 +1,5 @@ import cmd + import ps cmd.need("grub-mkimage") diff --git a/rwx/log/__init__.py b/rwx/log/__init__.py index a567492..3502ab5 100644 --- a/rwx/log/__init__.py +++ b/rwx/log/__init__.py @@ -1,5 +1,7 @@ import logging import sys + + def get_file_logger(name: str) -> logging.Logger: formatter = logging.Formatter( "%(name)s: %(asctime)s | %(levelname)s | %(filename)s:%(lineno)s | %(process)d >>> %(message)s", diff --git a/rwx/prj/sphinx.py b/rwx/prj/sphinx.py index a71bc35..ccb8d0f 100644 --- a/rwx/prj/sphinx.py +++ b/rwx/prj/sphinx.py @@ -1,4 +1,5 @@ from os import path + from sphinx.cmd.build import build_main from rwx.fs import wipe From a5ca3a60440d740a4d6b67e255c2d6c777a027c0 Mon Sep 17 00:00:00 2001 From: Marc Beninca Date: Mon, 10 Jun 2024 15:39:39 +0200 Subject: [PATCH 21/21] lint main --- rwx/__main__.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/rwx/__main__.py b/rwx/__main__.py index a857803..519fb8e 100755 --- a/rwx/__main__.py +++ b/rwx/__main__.py @@ -1,14 +1,11 @@ #! /usr/bin/env python3 - -import os - +from pathlib import Path import fs - if __name__ == "__main__": - file_path = os.path.realpath(__file__) - root_path = os.path.dirname(file_path) - directory_path = os.path.join(root_path, "tmp") - file_path = os.path.join(directory_path, "file") + file_path = Path(__file__).resolve() + root_path = file_path.parent + directory_path = root_path / "tmp" + file_path = directory_path / "file" fs.wipe(directory_path) fs.make_directory(directory_path)