ptrack

commit a045a0a935b12794a6b1261c0c08b04fae8fedcc

Author: Connor Etherington <[email protected]>

Date:   Sat Jul 29 06:02:15 2023 +0200

    Update.

---

 ptrack.egg-info/PKG-INFO                    |   6 +

 ptrack.egg-info/SOURCES.txt                 |   9 ++

 ptrack.egg-info/dependency_links.txt        |   1 +

 ptrack.egg-info/entry_points.txt            |   4 +

 ptrack.egg-info/requires.txt                |   3 +

 ptrack.egg-info/top_level.txt               |   1 +

 ptrack/__init__.py                          |  14 +++

 ptrack/__pycache__/__init__.cpython-310.pyc | Bin 0 -> 731 bytes

 ptrack/__pycache__/main.cpython-310.pyc     | Bin 0 -> 5541 bytes

 ptrack/main.py                              | 184 ++++++++++++++++++++++++++++

 setup.py                                    |  22 ++++

 11 files changed, 244 insertions(+)

diff --git a/ptrack.egg-info/PKG-INFO b/ptrack.egg-info/PKG-INFO

new file mode 100644

index 0000000..7da1525

--- /dev/null

+++ b/ptrack.egg-info/PKG-INFO

@@ -0,0 +1,6 @@

+Metadata-Version: 2.1

+Name: ptrack

+Version: 0.1.0

+Summary: A simple CLI utility for asthetically tracking progress when copying or moving files.

+Author: Connor Etherington

+Author-email: [email protected]

diff --git a/ptrack.egg-info/SOURCES.txt b/ptrack.egg-info/SOURCES.txt

new file mode 100644

index 0000000..f85fb63

--- /dev/null

+++ b/ptrack.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@

+setup.py

+ptrack/__init__.py

+ptrack/main.py

+ptrack.egg-info/PKG-INFO

+ptrack.egg-info/SOURCES.txt

+ptrack.egg-info/dependency_links.txt

+ptrack.egg-info/entry_points.txt

+ptrack.egg-info/requires.txt

+ptrack.egg-info/top_level.txt

diff --git a/ptrack.egg-info/dependency_links.txt b/ptrack.egg-info/dependency_links.txt

new file mode 100644

index 0000000..8b13789

--- /dev/null

+++ b/ptrack.egg-info/dependency_links.txt

@@ -0,0 +1 @@

+

diff --git a/ptrack.egg-info/entry_points.txt b/ptrack.egg-info/entry_points.txt

new file mode 100644

index 0000000..32a28e4

--- /dev/null

+++ b/ptrack.egg-info/entry_points.txt

@@ -0,0 +1,4 @@

+[console_scripts]

+ptc = ptrack.main:copy

+ptm = ptrack.main:move

+ptrack = ptrack.main:main

diff --git a/ptrack.egg-info/requires.txt b/ptrack.egg-info/requires.txt

new file mode 100644

index 0000000..382c1b3

--- /dev/null

+++ b/ptrack.egg-info/requires.txt

@@ -0,0 +1,3 @@

+rich

+argparse

+argcomplete

diff --git a/ptrack.egg-info/top_level.txt b/ptrack.egg-info/top_level.txt

new file mode 100644

index 0000000..c003217

--- /dev/null

+++ b/ptrack.egg-info/top_level.txt

@@ -0,0 +1 @@

+ptrack

diff --git a/ptrack/__init__.py b/ptrack/__init__.py

new file mode 100644

index 0000000..919fce7

--- /dev/null

+++ b/ptrack/__init__.py

@@ -0,0 +1,14 @@

+import argparse

+import argcomplete

+

+parser = argparse.ArgumentParser(description='A simple CLI utility for asthetically tracking progress when copying or moving files.')

+parser.add_argument('-v', '--verbose', action='store_true', help='verbose output')

+parser.add_argument('-c', '--copy', action='store_true', help='copy files (You can use `ptc` instead of `ptrack -c`)')

+parser.add_argument('-m', '--move', action='store_true', help='move files (You can use `ptm` instead of `ptrack -m`)')

+

+argcomplete.autocomplete(parser)

+args, unknown = parser.parse_known_args()

+

+verbose = args.verbose

+copy = args.copy

+move = args.move

diff --git a/ptrack/__pycache__/__init__.cpython-310.pyc b/ptrack/__pycache__/__init__.cpython-310.pyc

new file mode 100644

index 0000000..d544d60

Binary files /dev/null and b/ptrack/__pycache__/__init__.cpython-310.pyc differ

diff --git a/ptrack/__pycache__/main.cpython-310.pyc b/ptrack/__pycache__/main.cpython-310.pyc

new file mode 100644

index 0000000..c3479bd

Binary files /dev/null and b/ptrack/__pycache__/main.cpython-310.pyc differ

diff --git a/ptrack/main.py b/ptrack/main.py

new file mode 100755

index 0000000..bddf173

--- /dev/null

+++ b/ptrack/main.py

@@ -0,0 +1,184 @@

+import os

+import sys

+from rich.progress import Progress, BarColumn, TextColumn, TimeRemainingColumn, FileSizeColumn

+from rich.console import Console

+import shutil

+import ptrack

+

+verbose = ptrack.verbose

+argCopy = ptrack.copy

+argMove = ptrack.move

+

+

+def format_file_size(file_size):

+    if file_size >= 1024 * 1024 * 1024:

+        return f"{file_size / (1024*1024*1024):.2f} GB"

+    elif file_size >= 1024 * 1024:

+        return f"{file_size / (1024*1024):.2f} MB"

+    elif file_size >= 1024:

+        return f"{file_size / 1024:.2f} kB"

+    else:

+        return f"{file_size} bytes"

+

+

+def regular_copy(src, dst, console, task, progress):

+    with open(src, 'rb') as fsrc, open(dst, 'wb') as fdst:

+        while True:

+            buf = fsrc.read(1024*1024)

+            if not buf:

+                break

+            fdst.write(buf)

+            progress.update(task, advance=len(buf))

+            progress.refresh()

+

+

+def verbose_copy(src, dst, console, current, total_files):

+

+    file_size = os.path.getsize(src)

+

+    with Progress(

+        BarColumn(bar_width=50),

+        "[progress.percentage]{task.percentage:>3.0f}%",

+        TimeRemainingColumn(),

+        "[#ea2a6f][[/#ea2a6f]",

+        FileSizeColumn(),

+        "[#ea2a6f]/[/#ea2a6f]",

+        TextColumn(f"[bold cyan]{format_file_size(file_size)}[/bold cyan]"),

+        "[#ea2a6f]][/#ea2a6f]",

+        f"({current} of {total_files})",

+        console=console,

+        auto_refresh=False

+    ) as progress:

+        task = progress.add_task("", total=file_size, file_size=format_file_size(file_size))

+

+        with open(src, 'rb') as fsrc, open(dst, 'wb') as fdst:

+            while not progress.finished:

+                buf = fsrc.read(1024*1024)

+                if not buf:

+                    break

+                fdst.write(buf)

+                progress.update(task, advance=len(buf))

+                progress.refresh()

+

+

+def run(process):

+    console = Console()

+

+    if len(sys.argv) < 3:

+        hlp()

+        if process == "Copying":

+            console.print("[bold cyan]Usage: ptc [OPTIONS] SOURCE... DESTINATION[/bold cyan]")

+        elif process == "Moving":

+            console.print("[bold cyan]Usage: ptm [OPTIONS] SOURCE... DESTINATION[/bold cyan]")

+        sys.exit(1)

+

+    src_paths = sys.argv[1:-1]

+    dst = sys.argv[-1]

+    srcPaths = []

+

+    for path in src_paths:

+        if path.endswith('/'):

+            path = path[:-1]

+        srcPaths.append(path)

+

+    if os.path.isdir(dst):

+        dst_dir = dst

+        new_name = None

+    else:

+        dst_dir = os.path.dirname(dst)

+        new_name = os.path.basename(dst)

+

+    total_files = sum(len(files) for path in srcPaths for r, d, files in os.walk(path) if os.path.isdir(path)) + sum(1 for path in srcPaths if os.path.isfile(path))

+    total_size = sum(os.path.getsize(os.path.join(r, f)) for path in srcPaths for r, d, files in os.walk(path) for f in files) + sum(os.path.getsize(path) for path in srcPaths if os.path.isfile(path))

+    current_file = 1

+

+    if total_files > 1:

+        console.print(f"\n[#ea2a6f]{process}:[/#ea2a6f] [bold cyan]{total_files} files[/bold cyan]\n")

+    else:

+        for src_path in srcPaths:

+            if os.path.isfile(src_path):

+                console.print(f"\n[#ea2a6f]{process}:[/#ea2a6f] [bold cyan] {os.path.basename(src_path)} [/bold cyan]\n")

+

+    if verbose:

+        for src_path in srcPaths:

+            if os.path.isfile(src_path):

+                dst_path = os.path.join(dst_dir, os.path.basename(src_path) if not new_name else new_name)

+                verbose_copy(src_path, dst_path, console, current_file, total_files)

+                current_file += 1

+            else:

+                for root, dirs, files in os.walk(src_path):

+                    for file in files:

+                        src_file_path = os.path.join(root, file)

+                        relative_path = os.path.relpath(src_file_path, start=src_path)

+                        dst_file_path = os.path.join(dst_dir, os.path.basename(src_path) if not new_name else new_name, relative_path)

+                        os.makedirs(os.path.dirname(dst_file_path), exist_ok=True)

+                        verbose_copy(src_file_path, dst_file_path, console, current_file, total_files)

+                        current_file += 1

+    else:

+        with Progress(

+            BarColumn(bar_width=50),

+            "[progress.percentage]{task.percentage:>3.0f}%",

+            TimeRemainingColumn(),

+            "[#ea2a6f][[/#ea2a6f]",

+            FileSizeColumn(),

+            "[#ea2a6f]/[/#ea2a6f]",

+            TextColumn("[bold cyan]{task.fields[total_size]}[/bold cyan]"),

+            "[#ea2a6f]][/#ea2a6f]",

+            console=console,

+            auto_refresh=False

+        ) as progress:

+            task = progress.add_task("", total=total_size, total_size=format_file_size(total_size))

+

+            for src_path in srcPaths:

+                if os.path.isfile(src_path):

+                    dst_file_path = os.path.join(dst_dir, os.path.basename(src_path) if not new_name else new_name)

+                    regular_copy(src_path, dst_file_path, console, task, progress)

+                else:

+                    for root, dirs, files in os.walk(src_path):

+                        for file in files:

+                            src_file_path = os.path.join(root, file)

+                            relative_path = os.path.relpath(src_file_path, start=src_path)

+                            dst_file_path = os.path.join(dst_dir, os.path.basename(src_path) if not new_name else new_name, relative_path)

+                            os.makedirs(os.path.dirname(dst_file_path), exist_ok=True)

+                            regular_copy(src_file_path, dst_file_path, console, task, progress)

+

+    return srcPaths

+

+

+def copy():

+    run('Copying')

+

+

+def move():

+    src_paths = run('Moving')

+    for src_path in src_paths:

+        if os.path.isfile(src_path):

+            os.remove(src_path)

+        else:

+            shutil.rmtree(src_path)

+

+

+def hlp():

+    print("""

+usage: ptrack [-h] [-v] [-c] [-m]

+

+A simple CLI utility for asthetically tracking progress when copying or moving files.

+

+options:

+  -h, --help     show this help message and exit

+  -v, --verbose  verbose output

+  -c, --copy     copy files (You can use `ptc` instead of `ptrack -c`)

+  -m, --move     move files (You can use `ptm` instead of `ptrack -m`)

+""")

+

+

+def main():

+    if argMove:

+        move()

+    elif argCopy:

+        copy()

+    else:

+        hlp()

+

+if __name__ == "__main__":

+    main()

diff --git a/setup.py b/setup.py

new file mode 100644

index 0000000..fdea560

--- /dev/null

+++ b/setup.py

@@ -0,0 +1,22 @@

+from setuptools import setup, find_packages

+

+setup(

+    name='ptrack',

+    version='0.1.0',

+    description='A simple CLI utility for asthetically tracking progress when copying or moving files.',

+    author='Connor Etherington',

+    author_email='[email protected]',

+    packages=find_packages(),

+    install_requires=[

+        'rich',

+        'argparse',

+        'argcomplete',

+    ],

+    entry_points={

+        'console_scripts': [

+            'ptc=ptrack.main:copy',

+            'ptm=ptrack.main:move',

+            'ptrack=ptrack.main:main',

+        ]

+    }

+)