rclone-api 1.1.49__tar.gz → 1.1.50__tar.gz

{rclone_api-1.1.49 → rclone_api-1.1.50}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rclone_api
-Version: 1.1.49
+Version: 1.1.50
 Summary: rclone api in python
 Home-page: https://github.com/zackees/rclone-api
 License: BSD 3-Clause License

{rclone_api-1.1.49 → rclone_api-1.1.50}/pyproject.toml

@@ -23,7 +23,7 @@ dependencies = [
 ]
 
 # Change this with the version number bump.
-version = "1.1.49"
+version = "1.1.50"
 
 [tool.setuptools]
 package-dir = {"" = "src"}

{rclone_api-1.1.49 → rclone_api-1.1.50}/src/rclone_api/mount.py

@@ -1,9 +1,12 @@
+import atexit
 import os
 import platform
 import shutil
 import subprocess
 import time
 import warnings
+import weakref
+from concurrent.futures import ThreadPoolExecutor
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
@@ -12,6 +15,27 @@ from rclone_api.process import Process
 
 _SYSTEM = platform.system()  # "Linux", "Darwin", "Windows", etc.
 
+_MOUNTS_FOR_GC: weakref.WeakSet = weakref.WeakSet()
+
+
+def _add_mount_for_gc(mount: "Mount") -> None:
+    # weak reference to avoid circular references
+    _MOUNTS_FOR_GC.add(mount)
+
+
+def _remove_mount_for_gc(mount: "Mount") -> None:
+    _MOUNTS_FOR_GC.discard(mount)
+
+
+def _cleanup_mounts() -> None:
+    with ThreadPoolExecutor() as executor:
+        mount: Mount
+        for mount in _MOUNTS_FOR_GC:
+            executor.submit(mount.close, wait=False)
+
+
+atexit.register(_cleanup_mounts)
+
 
 @dataclass
 class Mount:
@@ -28,15 +52,18 @@ class Mount:
         assert isinstance(self.mount_path, Path)
         assert self.process is not None
         wait_for_mount(self.mount_path, self.process)
+        _add_mount_for_gc(self)
 
     def close(self, wait=True) -> None:
         """Clean up the mount."""
         if self._closed:
             return
         self._closed = True
+        self.process.terminate()
         clean_mount(self, verbose=False, wait=wait)
         if self.cache_dir and self.cache_dir_delete_on_exit:
             _cache_dir_delete_on_exit(self.cache_dir)
+        _remove_mount_for_gc(self)
 
     def __enter__(self) -> "Mount":
         return self
@@ -47,6 +74,10 @@ class Mount:
     def __del__(self):
         self.close(wait=False)
 
+    # make this a hashable object
+    def __hash__(self):
+        return hash(self.mount_path)
+
 
 def run_command(cmd: str, verbose: bool) -> int:
     """Run a shell command and print its output if verbose is True."""

{rclone_api-1.1.49 → rclone_api-1.1.50}/src/rclone_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rclone_api
-Version: 1.1.49
+Version: 1.1.50
 Summary: rclone api in python
 Home-page: https://github.com/zackees/rclone-api
 License: BSD 3-Clause License