karma365 0.2.0__py3-none-any.whl

karma365/__init__.py

@@ -0,0 +1,56 @@
+"""KARMA 365 SDK for Python. Your data stays on your phone; this talks to it."""
+
+from .client import AsyncKarmaClient, KarmaClient
+from .errors import (
+    AmbiguousNameError,
+    AppLockedError,
+    AppUpdateNeededError,
+    ConnectionLostError,
+    FileProblemError,
+    InvalidInputError,
+    KarmaError,
+    NotFoundError,
+    NotYourPhoneError,
+    ParentLockedError,
+    PhoneError,
+    PhoneNotFoundError,
+    PhoneStorageFullError,
+    PinLockedOutError,
+    PinRequiredError,
+    PinWrongError,
+    TokenError,
+    UploadInterruptedError,
+)
+from .link import Status
+from .models import InfoResult, MajorTopic, Resource, Task, Topic, WorkspaceStats
+
+__version__ = "0.2.0"
+__all__ = [
+    "AsyncKarmaClient",
+    "KarmaClient",
+    "Status",
+    "MajorTopic",
+    "Topic",
+    "Resource",
+    "Task",
+    "WorkspaceStats",
+    "InfoResult",
+    "KarmaError",
+    "TokenError",
+    "PhoneNotFoundError",
+    "NotYourPhoneError",
+    "AppLockedError",
+    "ConnectionLostError",
+    "AppUpdateNeededError",
+    "PinRequiredError",
+    "PinWrongError",
+    "PinLockedOutError",
+    "ParentLockedError",
+    "NotFoundError",
+    "AmbiguousNameError",
+    "InvalidInputError",
+    "FileProblemError",
+    "PhoneStorageFullError",
+    "UploadInterruptedError",
+    "PhoneError",
+]

karma365/__main__.py

@@ -0,0 +1,76 @@
+"""`python -m karma365 status` and `python -m karma365 speedtest`."""
+
+import asyncio
+import os
+import sys
+import tempfile
+import time
+
+from .client import AsyncKarmaClient
+from .errors import KarmaError
+from .upload import cancel_file, send_file
+
+
+async def status() -> int:
+    async with AsyncKarmaClient() as karma:
+        result = await karma.status()
+    if result.connected:
+        print(f"Connected to your phone at {result.phone_url} ({result.latency_ms} ms)")
+        return 0
+    print(f"Not connected: {result.message}")
+    if result.hint:
+        print(f"What to do: {result.hint}")
+    print(f"Code: {result.reason}")
+    return 1
+
+
+async def speedtest(megabytes: int) -> int:
+    """Sends a throwaway file over the real path and measures it. Wi-Fi, the phone's
+    decrypting and its storage all count, so this is the speed you will really get."""
+    async with AsyncKarmaClient() as karma:
+        minors = await karma.list_minors()
+        if not minors:
+            print("Create a minor topic first. The test needs one, and leaves nothing in it.")
+            return 1
+        handle, path = tempfile.mkstemp(suffix=".bin", prefix="karma_speedtest_")
+        try:
+            with os.fdopen(handle, "wb") as file:
+                for _ in range(megabytes):
+                    file.write(os.urandom(1024 * 1024))
+
+            def progress(sent: int, total: int) -> None:
+                print(f"\r{sent * 100 // total}%", end="", flush=True)
+
+            started = time.perf_counter()
+            resource_id = await send_file(
+                karma._link,
+                path=path,
+                file_type="video",
+                topic_id=minors[0].id,
+                title="KARMA 365 speed test",
+                on_progress=progress,
+            )
+            seconds = time.perf_counter() - started
+            await karma.delete(resource_id)
+            print(f"\n{megabytes} MB in {seconds:.1f} s: {megabytes / seconds:.1f} MB/s")
+            return 0
+        finally:
+            os.unlink(path)
+
+
+def main() -> int:
+    command = sys.argv[1] if len(sys.argv) > 1 else "status"
+    try:
+        if command == "status":
+            return asyncio.run(status())
+        if command == "speedtest":
+            return asyncio.run(speedtest(int(sys.argv[2]) if len(sys.argv) > 2 else 100))
+    except KarmaError as error:
+        print(error)
+        return 1
+    print("Usage: python -m karma365 status\n       python -m karma365 speedtest [MB]")
+    return 2
+
+
+if __name__ == "__main__":
+    sys.exit(main())