dmart 1.4.39__py3-none-any.whl → 1.4.40.post1__py3-none-any.whl

dmart/cli.py

@@ -0,0 +1,1103 @@
+#!/usr/bin/env python
+
+import os
+from prompt_toolkit import PromptSession
+import sys
+
+from prompt_toolkit.filters import has_completions, completion_is_selected
+from prompt_toolkit.key_binding import KeyBindings
+from rich import pretty
+from rich import print
+from rich.console import Console
+from pydantic_settings import BaseSettings, SettingsConfigDict
+import requests
+import re
+
+# from rich.tree import Tree
+from rich.table import Table
+from dataclasses import dataclass
+from prompt_toolkit.styles import Style
+from prompt_toolkit.formatted_text import HTML
+from prompt_toolkit.completion import Completer, Completion  # , FuzzyCompleter
+from prompt_toolkit.history import FileHistory
+from prompt_toolkit.cursor_shapes import CursorShape
+from rich.traceback import install
+import json
+import pathlib
+from enum import Enum
+import termios
+import tty
+from dataclasses import field
+from pathlib import Path
+
+install(show_locals=False)
+
+KEYWORDS = [
+    "ls",
+    "switch",
+    "mv",
+    "cat",
+    "cd",
+    "rm",
+    "print",
+    "pwd",
+    "help",
+    "mkdir",
+    "schema",
+    "create",
+    "attach",
+    "request",
+    "csv",
+    "progress",
+]
+
+style = Style.from_dict(
+    {
+        "completion-menu.completion": "bg:#008888 #ffffff",
+        "completion-menu.completion.current": "bg:#00aaaa #000000",
+        "scrollbar.background": "bg:#88aaaa",
+        "scrollbar.button": "bg:#222222",
+        "prompt": "#dddd22 bold",
+    }
+)
+
+console = Console()
+pretty.install()
+
+
+class Settings(BaseSettings):
+    url: str = "http://localhost:8282"
+    shortname: str = "dmart"
+    password: str = "password"
+    query_limit: int = 100
+    retrieve_json_payload: bool = True
+    default_space: str = "management"
+    pagination: int = 5
+
+    model_config = SettingsConfigDict(env_file = os.getenv("BACKEND_ENV", os.path.dirname(os.path.realpath(__file__)) + "/config.ini"), env_file_encoding = "utf-8")
+
+
+settings = Settings()
+
+
+class CLI_MODE(str, Enum):
+    REPL = "REPL"
+    CMD = "CMD"
+    SCRIPT = "SCRIPT"
+
+
+mode = CLI_MODE.REPL
+
+
+class SpaceManagmentType(str, Enum):
+    CREATE = "create"
+    UPDATE = "update"
+    DELETE = "delete"
+
+
+@dataclass
+class DMart:
+    session = requests.Session()
+    headers = {"Content-Type": "application/json"}
+    dmart_spaces : list = field(default_factory=lambda: [])
+    space_names : list[str] = field(default_factory=list[str])
+    current_space: str = settings.default_space
+    current_subpath: str = "/"
+    current_subpath_entries : list = field(default_factory=list)
+
+    def __dmart_api(self, endpoint, json=None):
+        url = f"{settings.url}{endpoint}"
+
+        if json:
+            response = self.session.post(url, headers=self.headers, json=json)
+        else:
+            response = self.session.get(url)
+
+        response_body = response.json()
+        if response.status_code != 200:
+            print(f"[red]{endpoint}", end="\r")
+        return response_body
+
+    def create_content(self, endpoint, json):
+        url = f"{settings.url}{endpoint}"
+
+        response = self.session.post(url, headers=self.headers, json=json)
+
+        if response.status_code != 200:
+            print(endpoint, response.json())
+        return response.json().get('status', None)
+
+    def delete(self, spacename, subpath, shortname, resource_type):
+        endpoint = "/managed/request"
+        json = {
+            "space_name": spacename,
+            "request_type": "delete",
+            "records": [
+                {
+                    "resource_type": resource_type,
+                    "subpath": subpath,
+                    "shortname": shortname,
+                    "attributes": {},
+                }
+            ],
+        }
+        return self.__dmart_api(endpoint, json)
+
+    def register(self, invitation):
+        json = {
+            "resource_type": "user",
+            "spacename": "management",
+            "subpath": "/users",
+            "shortname": settings.shortname,
+            "attributes": {"invitation": invitation, "password": settings.password},
+        }
+        self.__dmart_api("/user/create", json)
+
+    def login(self):
+        json = {"shortname": settings.shortname, "password": settings.password}
+        response = self.__dmart_api("/user/login", json)
+        if response["status"] == "success":
+            self.headers = {
+                **self.headers,
+                "Authorization": f'Bearer {response["records"][0]["attributes"]["access_token"]}',
+            }
+        return response
+
+    def profile(self):
+        self.__dmart_api("/user/profile")
+
+    def spaces(self, force: bool = False):
+        if force or not self.dmart_spaces:
+            json = {
+                "type": "spaces",
+                "space_name": "management",
+                "subpath": "/",
+            }
+
+            response = self.__dmart_api("/managed/query", json)
+            self.dmart_spaces = response["records"]
+            self.space_names = [one["shortname"] for one in self.dmart_spaces]
+        return self.dmart_spaces
+
+    def query(self, json):
+        json["limit"] = settings.query_limit
+        json["retrieve_json_payload"] = settings.retrieve_json_payload
+        return self.__dmart_api("/managed/query", json)
+
+    def meta(self, resource_type, space_name, subpath, shortname):
+        endpoint = "managed/meta"
+        url = f"{settings.url}/{endpoint}/{resource_type}/{space_name}/{subpath}/{shortname}"
+        response = self.session.get(url)
+
+        if response.status_code != 200:
+            print(endpoint, response.json())
+        return response.json()
+
+    def payload(self, resource_type, space_name, subpath, shortname):
+        endpoint = "managed/payload"
+        url = f"{settings.url}/{endpoint}/{resource_type}/{space_name}/{subpath}/{shortname}.json"
+        response = self.session.get(url)
+
+        # if response.status_code != 200:
+        #    print(endpoint, response.json())
+        return response.status_code, response.json()
+
+    def print_spaces(self):
+        print_spaces = "Available spaces: "
+        for one in self.space_names:
+            if self.current_space == one:
+                one = f" [bold yellow]{one}[/] "
+            else:
+                one = f" [blue]{one}[/] "
+            print_spaces += one
+        print(print_spaces)
+
+    def list(self):
+        json = {
+            "space_name": dmart.current_space,
+            "type": "subpath",
+            "subpath": dmart.current_subpath.replace("//", "/"),
+            "retrieve_json_payload": True,
+            "limit": 100,
+        }
+
+        ret = self.query(json)
+        self.current_subpath_entries.clear()
+        if "records" in ret:
+            for one in ret["records"]:
+                self.current_subpath_entries.append(one)
+
+    def get_mime_type(self, ext):
+        match ext:
+            case ".jfif" | ".jfif-tbnl" | ".jpe" | ".jpeg" | ".jpg":
+                return "image/jpeg"
+            case ".png":
+                return "image/png"
+            case ".json":
+                return "application/json"
+
+    def create_folder(self, shortname):
+        json = {
+            "space_name": self.current_space,
+            "request_type": "create",
+            "records": [
+                {
+                    "resource_type": "folder",
+                    "subpath": self.current_subpath,
+                    "shortname": shortname,
+                    "attributes": {"is_active": True},
+                }
+            ],
+        }
+        endpoint = "/managed/request"
+        return self.__dmart_api(endpoint, json)
+
+    def manage_space(self, spacename, mode: SpaceManagmentType):
+        endpoint = "/managed/space"
+        json = {
+            "space_name": spacename,
+            "request_type": mode.value,
+            "records": [
+                {
+                    "resource_type": "space",
+                    "subpath": "/",
+                    "shortname": spacename,
+                    "attributes": {},
+                }
+            ],
+        }
+        result = self.__dmart_api(endpoint, json)
+        self.spaces(force=True)
+        return result
+
+    def create_entry(self, shortname, resource_type):
+        endpoint = "/managed/request"
+        json = {
+            "space_name": self.current_space,
+            "request_type": "create",
+            "records": [
+                {
+                    "resource_type": resource_type,
+                    "subpath": self.current_subpath,
+                    "shortname": shortname,
+                    "attributes": {"is_active": True},
+                }
+            ],
+        }
+        return self.__dmart_api(endpoint, json)
+
+    def create_attachments(self, request_record: dict, payload_file):
+        endpoint = f"{settings.url}/managed/resource_with_payload"
+        with open("temp.json", "w") as request_record_file:
+            json.dump(request_record, request_record_file)
+
+        with open("temp.json", "rb") as request_file:
+            with open(payload_file, "rb") as media_file:
+                request_file.seek(0)
+                data = [
+                    (
+                        "request_record",
+                        ("record.json", request_file, "application/json"),
+                    ),
+                    (
+                        "payload_file",
+                        (
+                            media_file.name.split("/")[-1],
+                            media_file,
+                            self.get_mime_type(pathlib.Path(payload_file).suffix),
+                        ),
+                    ),
+                ]
+                response = self.session.post(
+                    endpoint,
+                    data={"space_name": self.current_space},
+                    files=data,
+                    headers=self.headers,
+                )
+                if response.status_code != 200:
+                    print(endpoint, response.json())
+
+                os.remove("temp.json")
+                return response.json()
+
+    def upload_csv(self, resource_type, subpath, schema_shortname, payload_file):
+        with open(payload_file, "rb") as media_file:
+            endpoint = f"{settings.url}/managed/resources_from_csv/{resource_type}/{dmart.current_space}/{subpath}/{schema_shortname}"
+            headers = {**self.headers}
+            del headers["Content-Type"]
+            response = self.session.post(
+                endpoint,
+                files=[
+                    ('resources_file', ('file', media_file, 'text/csv'))
+                ],
+                headers=headers,
+            )
+            if response.status_code != 200:
+                print(endpoint, response.json())
+
+            return response.json()
+
+    def upload_schema(self, shortname, payload_file_path):
+        data = {
+            "resource_type": "schema",
+            "subpath": "schema",
+            "shortname": shortname,
+            "attributes": {"schema_shortname": "meta_schema", "is_active": True},
+        }
+        endpoint = f"{settings.url}/managed/resource_with_payload"
+
+        with open("temp.json", "w") as request_record_file:
+            json.dump(data, request_record_file)
+
+        with open("temp.json", "rb") as request_record:
+            with open(payload_file_path, "rb") as payload_file:
+                files = {
+                    "space_name": (None, self.current_space),
+                    "request_record": request_record,
+                    "payload_file": payload_file,
+                }
+                headers = {**self.headers}
+                del headers["Content-Type"]
+                response = self.session.post(
+                    endpoint,
+                    files=files,
+                    headers=headers,
+                )
+
+                if response.status_code != 200:
+                    print(endpoint, response.json())
+
+                return response.json()
+
+    def upload_folder(self, folder_schema_path):
+        endpoint = f"{settings.url}/managed/request"
+        with open(folder_schema_path, "r") as folder_schema:
+            response = self.session.post(
+                url=endpoint,
+                data=folder_schema,
+                headers=self.headers,
+            )
+            if response.status_code != 200:
+                print(endpoint, response.json())
+            else:
+                print(response.json())
+        return response.json()
+
+    def move(
+        self,
+        resource_type,
+        source_path,
+        source_shortname,
+        destination_path,
+        destination_shortname,
+    ):
+        endpoint = "/managed/request"
+        data = {
+            "space_name": self.current_space,
+            "request_type": "move",
+            "records": [
+                {
+                    "resource_type": resource_type,
+                    "subpath": self.current_subpath,
+                    "shortname": source_shortname,
+                    "attributes": {
+                        "src_subpath": source_path,
+                        "src_shortname": source_shortname,
+                        "dest_subpath": destination_path,
+                        "dest_shortname": destination_shortname,
+                    },
+                }
+            ],
+        }
+        return self.__dmart_api(endpoint, data)
+
+    def import_zip(self, zip_file_path):
+        endpoint = f"{settings.url}/managed/import"
+        with open(zip_file_path, "rb") as zip_file:
+            headers = {**self.headers}
+            del headers["Content-Type"]
+            response = self.session.post(
+                endpoint,
+                files={"zip_file": zip_file},
+                headers=headers,
+            )
+            if response.status_code != 200:
+                print(endpoint, response.json())
+            return response.json()
+
+    def export_json(self, query_json_file):
+        output_file_name = "/export.zip"
+        os_downloads_path = str(Path.home()/"Downloads")
+        output_file_path = os_downloads_path + output_file_name
+        with open(query_json_file, "r") as f:
+            data = json.load(f)
+        endpoint = f"{settings.url}/managed/export"
+        response = self.session.post(endpoint, headers=self.headers, json=data, stream=True)
+        if response.status_code != 200:
+            print(endpoint, response.json())
+            return response.json()
+        with open(output_file_path, "wb") as out_file:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    out_file.write(chunk)
+        print(f"Exported to {os_downloads_path}")
+        return {"status": "success", "file": output_file_path}
+
+
+dmart = DMart()
+
+
+class CustomCompleter(Completer):
+    def get_completions(self, document, complete_event):
+        cmd = document.text
+        arg = document.get_word_under_cursor()
+        if len(cmd) < 2 or re.match(r"^\s*$", cmd):
+            for one in KEYWORDS:
+                if one.startswith(arg):
+                    yield Completion(one, start_position=-len(arg))
+        elif re.match(r"^\s*\w+\s+", cmd):
+            if cmd.startswith("s"):
+                for one in dmart.space_names:
+                    if one.startswith(arg):
+                        yield Completion(one, start_position=-len(arg))
+            else:
+                for one in dmart.current_subpath_entries:
+                    if one[ "shortname" ].startswith(
+                        arg
+                    ):
+                        if "cd" in cmd and one["resource_type"] == "folder":
+                            yield Completion(
+                                one["shortname"],
+                                start_position=-len(arg),
+                                display_meta=one["resource_type"],
+                            )
+                        elif "cd" not in cmd:
+                            yield Completion(
+                                one["shortname"],
+                                start_position=-len(arg),
+                                display_meta=one["resource_type"],
+                            )
+
+
+
+def bottom_toolbar():
+    return HTML(
+        f'<b>{dmart.current_space}</b> : <b><style bg="ansired">{dmart.current_subpath}</style></b>'
+    )
+
+
+def check_update_space(space, subpath=None):
+    for one in dmart.space_names:
+        if one.startswith(space) and not one.startswith(dmart.current_space):
+            dmart.current_space = one
+            print("Current space switched to:", dmart.current_space)
+            if subpath is not None:
+                dmart.current_subpath = subpath
+            dmart.list()
+            dmart.print_spaces()
+            break
+
+
+def action(text: str):
+    old_space, old_subpath = dmart.current_space, dmart.current_subpath
+    match text.split():
+        case ["h" | "help" | "?"]:
+            table = Table(title="Help")
+            table.add_column("Command")
+            table.add_column("Description")
+            table.add_row(
+                r"[blue]s[paces][/] [green]\[space_name][/]",
+                "List availble spaces or switch to space",
+            )
+            table.add_row(
+                r"[blue]ls[/] [green]\[folder_name][/]",
+                "List entries under current subpath",
+            )
+            table.add_row("[blue]pwd[/]", "Print current subpath")
+            table.add_row("[blue]cd[/] [green]folder_name[/]", "Enter the folder")
+            table.add_row("[blue]cd ..[/]", "Go one-level up with the subpath")
+            table.add_row(
+                r"[blue]p\[rint][/blue] [green]shortname[/]",
+                "Print meta data for the entry shortname under current subpath",
+            )
+            table.add_row(
+                r"[blue]c\[at][/] [green]shortname[/]",
+                "Print the json payload (only) for th entry shortnae under current subpath",
+            )
+            table.add_row(
+                "[blue]rm[/] [green]shortname|*[/]",
+                "Delete the shortname (entry or attachment)",
+            )
+            table.add_row(
+                "[blue]create [space|folder] <shortname>[/]", "Create space/folder for current space"
+            )
+            table.add_row(
+                "[blue]upload csv <resource_type> <shortname> <schema_shortname> <csv_file>[/]",
+                "Upload data to the current space with schema validation",
+            )
+            table.add_row(
+                "[blue]upload schema <shortname> <json_file>[/]",
+                "Upload schema to the current space",
+            )
+            table.add_row(
+                "[blue]request <resource_type> <json_file>[/]",
+                "Add/Manage a resource to/in the space",
+            )
+            table.add_row(
+                "[blue]progress <subpath> <shortname> <action>[/]",
+                "Progress a ticket into a new state using an action",
+            )
+            table.add_row(
+                "[blue]move <resource_type> <source> <destination>[/]", "Move resource"
+            )
+            table.add_row("[blue]import <zip_file>[/]", "Import a ZIP file")
+            table.add_row("[blue]export <query_json> [/]", "Export a ZIP file")
+            table.add_row("[blue]exit|Ctrl+d[/]", "Exit the app")
+            table.add_row("[blue]help|h|?[/]", "Show this help")
+            print(table)
+            return None
+        case ["mkdir", dir_shortname]:
+            print(dmart.create_folder(dir_shortname))
+            return None
+        case ["attach", *args]:
+            shortname = None
+            entry_shortname = None
+            payload_file = None
+            payload_type = None
+
+            shortname = args[0]
+            entry_shortname = args[1]
+            payload_type = args[2]
+            payload_file = args[3]
+
+            print(
+                dmart.create_attachments(
+                    {
+                        "shortname": shortname,
+                        "resource_type": payload_type,
+                        "subpath": f"{dmart.current_subpath}/{entry_shortname}",
+                        "attributes": {"is_active": True},
+                    },
+                    payload_file,
+                )
+            )
+            return None
+        case ["upload", "schema", *args]:
+            shortname = None
+            payload_file = None
+
+            if len(args) == 3:
+                search = re.search(r"@\w+", args[2])
+                if not search:
+                    print("[red]Malformated Command")
+                    return None
+                space = search.group()
+                space = space.replace("@", "")
+                check_update_space(space)
+                dmart.current_subpath = args[2].replace(f"@{space}/", "")
+                dmart.list()
+            shortname = args[0]
+            payload_file = args[1]
+
+            dmart.upload_schema(shortname, payload_file)
+            check_update_space(old_space)
+            dmart.current_subpath = old_subpath
+            dmart.list()
+            return None
+        case ["create", *args]:
+            shortname = None
+            resource_type = None
+            if len(args) == 3:
+                search = re.search(r"@\w+", args[2])
+                if not search:
+                    print("[red]Malformated Command")
+                    return None
+                space = search.group()
+                space = space.replace("@", "")
+                check_update_space(space)
+                dmart.current_subpath = args[2].replace(f"@{space}", "")
+                dmart.list()
+            shortname = args[0]
+            resource_type = args[1]
+
+            if args[0] == "space":
+                print(dmart.manage_space(args[1], SpaceManagmentType.CREATE))
+            else:
+                print(dmart.create_entry(resource_type, shortname))
+
+            check_update_space(old_space)
+            dmart.current_subpath = old_subpath
+            dmart.list()
+            return None
+        case ["move", type, source, destination]:
+            if not source.startswith("/"):
+                source += f"{dmart.current_subpath}/{source}"
+            if not destination.startswith("/"):
+                destination += f"{dmart.current_subpath}/{destination}"
+            source_path = "/".join(source.split("/")[:-1])
+            source_shortname = source.split("/")[-1]
+            destination_path = "/".join(destination.split("/")[:-1])
+            destination_shortname = destination.split("/")[-1]
+            print(
+                dmart.move(
+                    type,
+                    source_path,
+                    source_shortname,
+                    destination_path,
+                    destination_shortname,
+                )
+            )
+            return None
+
+        case ["progress", ticket_subpath, ticket_shortname, new_state]:
+            endpoint = f"{settings.url}/managed/progress-ticket/{dmart.current_space}/{ticket_subpath}/{ticket_shortname}/{new_state}"
+            response = dmart.session.put(
+                url=endpoint,
+                data={},
+                headers=dmart.headers,
+            )
+            print(response.json())
+            return None
+        case ["request", *args]:
+            if len(args) != 1:
+                print("[red]Malformated Command")
+                return None
+
+            with open(args[0]) as f:
+                return print(
+                dmart.create_content("/managed/request", json.load(f)), end="\r"
+                )
+
+            check_update_space(old_space)
+            dmart.current_subpath = old_subpath
+            dmart.list()
+            return None
+        case ["query", shortname]:
+            action(f"query {shortname}")
+            action("ls")
+            action("cd ..")
+            return None
+        case ["upload", "csv", *args]:
+            if len(args) == 4:
+                print(dmart.upload_csv(args[0], args[1], args[2], args[3]))
+            elif len(args) == 3:
+                print("[red]Malformated Command")
+                return None
+            check_update_space(old_space)
+            dmart.current_subpath = old_subpath
+            dmart.list()
+            return None
+        case ["rm", "*"]:
+            dmart.list()
+            for one in dmart.current_subpath_entries:
+                shortname = one["shortname"]
+                resource_type = one["resource_type"]
+                if shortname and resource_type:
+                    print(
+                        shortname,
+                        dmart.delete(
+                            dmart.current_space,
+                            dmart.current_subpath,
+                            shortname,
+                            resource_type,
+                        ),
+                    )
+            return None
+        case ["rm", *content]:
+            if content[0] == "space":
+                print(dmart.manage_space(content[1], SpaceManagmentType.DELETE))
+            else:
+                content = content[0]
+                if content.startswith("@"):
+                    path = content[1]
+                    search = re.search(r"@\w+", content)
+                    if not search:
+                        print("[red]Malformated Command")
+                        return None
+                    space = search.group()
+                    space = space.replace("@", "")
+                    check_update_space(space)
+                    dmart.list()
+                    content = content.replace(f"@{space}/", "")
+                # print(dmart.current_subpath_entries)
+                shortname = ""
+                resource_type = ""
+                for one in dmart.current_subpath_entries:
+                    if one["shortname"] == content:
+                        shortname = one["shortname"]
+                        resource_type = one["resource_type"]
+                if shortname and resource_type:
+                    print(
+                        dmart.delete(
+                            dmart.current_space,
+                            dmart.current_subpath,
+                            shortname,
+                            resource_type,
+                        )
+                    )
+                else:
+                    print("item not found")
+                check_update_space(old_space)
+                dmart.list()
+        case ["pwd"]:
+            print(f"{dmart.current_space}:{dmart.current_subpath}")
+            return None
+        case ["cd"]:
+            dmart.current_subpath = "/"
+            dmart.list()
+            print(f"[yellow]Switched subpath to:[/] [green]{dmart.current_subpath}[/]")
+            return None
+        case ["cd", ".."]:
+            if dmart.current_subpath != "/":
+                dmart.current_subpath = "/".join(dmart.current_subpath.split("/")[:-1])
+            if not dmart.current_subpath:
+                dmart.current_subpath = "/"
+            dmart.list()
+            print(f"[yellow]Switched subpath to:[/] [green]{dmart.current_subpath}[/]")
+            return None
+        case ["p" | "print", content]:
+            shortname = ""
+            resource_type = ""
+            shortname = content
+            for one in dmart.current_subpath_entries:
+                if one["shortname"].startswith(shortname):
+                    shortname = one["shortname"]
+                    resource_type = one["resource_type"]
+            if shortname and resource_type:
+                print(
+                    dmart.meta(
+                        resource_type,
+                        dmart.current_space,
+                        dmart.current_subpath,
+                        shortname,
+                    )
+                )
+            else:
+                print("item not found")
+        case ["cd", directory]:
+            if directory.startswith("@"):
+                search = re.search(r"@\w+", directory)
+                if not search:
+                    print("[red]Malformated Command")
+                    return None
+                space = search.group()
+                space = space.replace("@", "")
+                check_update_space(space)
+                dmart.current_subpath = directory.replace(f"@{space}/", "")
+                dmart.list()
+                directory = dmart.current_subpath
+            new_subpath = ""
+            for one in dmart.current_subpath_entries:
+                if (
+                    one["shortname"].startswith(directory)
+                    and one["resource_type"] == "folder"
+                ):
+                    new_subpath = (
+                        ""
+                        if dmart.current_subpath == "/" or dmart.current_subpath == ""
+                        else dmart.current_subpath
+                    )
+                    if new_subpath != "":
+                        new_subpath += "/"
+                    new_subpath += one["shortname"]
+                    dmart.current_subpath = new_subpath
+                    dmart.list()
+                    print(
+                        f"[yellow]Switched subpath to:[/] [green]{dmart.current_subpath}[/]"
+                    )
+                    break
+                    return None
+            return None
+        case ["c" | "cat", *extra_shortname]:
+            old_path = dmart.current_subpath
+            if "/" in extra_shortname[0]:
+                dmart.current_subpath = "/".join(extra_shortname[0].split("/")[:-1])
+                extra_shortname[0] = "".join(extra_shortname[0].split("/")[-1])
+            dmart.list()
+            dmart.current_subpath = old_path
+
+            record = {}
+            if extra_shortname[0] == "*":
+                for one in dmart.current_subpath_entries:
+                    print(one)
+            elif len(extra_shortname) > 0:
+                shortname = extra_shortname[0]
+                for one in dmart.current_subpath_entries:
+                    if one["shortname"].startswith(shortname):
+                        record = one
+            if record is not None:
+                print(record)
+            else:
+                print("[yellow]Item is not found[/]")
+        case ["ls", *_extra_subpath]:
+            if len(_extra_subpath) >= 2:
+                print("Too many args passed !")
+                return None
+
+            extra_subpath = ""
+            if len(_extra_subpath) == 1 and not _extra_subpath[0].isnumeric():
+                if _extra_subpath[0].startswith("/"):
+                    _extra_subpath[0] = _extra_subpath[0][1:]
+                if _extra_subpath[0].endswith("/"):
+                    _extra_subpath[0] = _extra_subpath[0][:-1]
+
+                if _extra_subpath[0].startswith("@"):
+                    search = re.search(r"@\w+", _extra_subpath[0])
+                    if not search:
+                        print("[red]Malformated Command")
+                        return None
+                    space = search.group()
+                    space = space.replace("@", "")
+                    check_update_space(space)
+                    dmart.current_subpath = _extra_subpath[0].replace(f"@{space}/", "")
+                    dmart.list()
+                else:
+                    dmart.current_subpath = _extra_subpath[0]
+                    dmart.list()
+
+
+            if extra_subpath == "":
+                for one in dmart.current_subpath_entries:
+                    if one["shortname"].startswith(extra_subpath):
+                        extra_subpath = one["shortname"]
+
+            path = f"[yellow]{dmart.current_space}[/]:[blue]{dmart.current_subpath}"
+            if dmart.current_subpath != "/":
+                path = f"{path}/{extra_subpath}"
+            else:
+                path = f"{path}{extra_subpath}"
+            # tree = Tree(path)
+
+            pagination_length = 0
+            if len(_extra_subpath) >= 1 and _extra_subpath[-1].isnumeric():
+                pagination_length = int(_extra_subpath[-1])
+            else:
+                pagination_length = settings.pagination
+
+            pagination_bucket : list[list] = []
+            bucket = []
+            idx = 0
+            for one in dmart.current_subpath_entries:
+                icon = ":page_facing_up:"
+                extra = ""
+                if one["resource_type"] == "folder":
+                    icon = ":file_folder:"
+                if (
+                    isinstance(one, dict)
+                    and "attributes" in one
+                    and isinstance(one["attributes"], dict)
+                    and "payload" in one["attributes"]
+                    and isinstance(one["attributes"]["payload"], dict)
+                    and "content_type" in one["attributes"]["payload"]
+                ):
+                    if "schema_shortname" in one["attributes"]["payload"]:
+                        schema = f",schema= {one["attributes"]["payload"].get("schema_shortname", "N/A")}"
+                        extra = f"[yellow](payload:type={one['attributes']['payload']['content_type']}{schema})[/]"
+
+                idx += 1
+                bucket.append(f"{icon} [green]{one['shortname']}[/] {extra}")
+                if idx == pagination_length:
+                    idx = 0
+                    pagination_bucket.append(bucket)
+                    bucket = []
+
+            if len(bucket) != 0:
+                pagination_bucket.append(bucket)
+            idx = 0
+            c = ""
+            if len(pagination_bucket) == 1:
+                for bucket in pagination_bucket[0]:
+                    print(bucket)
+            else:
+                while True:
+                    if len(pagination_bucket) == 0:
+                        break
+
+                    for bucket in pagination_bucket[idx]:
+                        print(bucket)
+
+                    if idx >= len(pagination_bucket) - 1:
+                        break
+
+                    print("q: quite, n: next")
+                    fd = sys.stdin.fileno()
+                    old_settings = termios.tcgetattr(fd)
+                    try:
+                        tty.setraw(fd)
+                        c = sys.stdin.readline(1)
+
+                        if c == "q":
+                            break
+                        if c == "n" or c == "\r":
+                            idx += 1
+                    finally:
+                        termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+                        return None
+                # c = input("q: quite, b: previous, n: next = ")
+            # print(tree)
+            dmart.current_space, dmart.current_subpath = old_space, old_subpath
+            dmart.list()
+            return None
+        case ["s" | "switch", *space]:
+            if len(space) == 0:
+                dmart.print_spaces()
+                return None
+            match = False
+            for one in dmart.space_names:
+                if space and one.startswith(space[0]):
+                    print(
+                        f"Switching current space from {dmart.current_space} to {one} / {space}"
+                    )
+                    dmart.current_space = one
+                    dmart.list()
+                    match = True
+                    dmart.print_spaces()
+                    break
+            if not match:
+                print(f"Requested space {space} not found")
+                return None
+            return None
+        case ["import", zip_file]:
+            print(dmart.import_zip(zip_file))
+            return None
+        case ["export", query_json_file]:
+            print(dmart.export_json(query_json_file))
+            return None
+        case _:
+            print(f"[red]Command[/] [yello]{text}[/] [red]unknown[/]")
+            return None
+
+
+var : dict = {}
+
+
+def parsing_variables(sliced_command):
+    for i in range(len(sliced_command)):
+        search = re.search(r"\$\w*", sliced_command[i])
+        if search is not None:
+            if v := search[0]:
+                sliced_command[i] = sliced_command[i].replace(v, var.get(v, ""))
+    return sliced_command
+
+
+key_bindings = KeyBindings()
+filter = has_completions & ~completion_is_selected
+@key_bindings.add("enter", filter=filter)
+def _(event):
+    event.current_buffer.go_to_completion(0)
+    event.current_buffer.validate_and_handle()
+
+
+if __name__ == "__main__":
+    try :
+        # print(Panel.fit("For help, type : [bold]?[/]", title="DMart Cli"))
+        print("[bold][green]DMART[/] [yellow]Command line interface[/][/]")
+        print(
+            f"Connecting to [yellow]{settings.url}[/] user: [yellow]{settings.shortname}[/]"
+        )
+
+        ret = dmart.login()
+        if ret["status"] == "failed":
+            print("Login failed")
+            exit(1)
+        dmart.profile()
+        spaces = dmart.spaces()
+        dmart.current_space = settings.default_space  # dmart.space_names[0]
+
+        dmart.list()
+        # print("Available spaces:", space_names)
+        # print("Current space:",  current_space)
+        dmart.print_spaces()
+        print("[red]Type [bold]?[/] for help[/]")
+        # current_subpath = "/"
+        # session = PromptSession(lexer=PygmentsLexer(DmartLexer), completer=dmart_completer, style=style)
+
+        session = PromptSession(
+            style=style,
+            completer=CustomCompleter(),
+            history=FileHistory(".cli_history"),
+            key_bindings=key_bindings
+        )
+
+        if len(sys.argv) >= 2:
+            if sys.argv[1] == "s":
+                mode = CLI_MODE.SCRIPT
+            elif sys.argv[1] == "c":
+                mode = CLI_MODE.CMD
+                if sys.argv[3].startswith("/"):
+                    check_update_space(sys.argv[2], sys.argv[3])
+                    del sys.argv[2]
+                    del sys.argv[2]
+                else:
+                    check_update_space(sys.argv[2])
+                    del sys.argv[2]
+            del sys.argv[0]
+            del sys.argv[0]
+
+        if mode == CLI_MODE.CMD:
+            print(sys.argv)
+            action(" ".join(sys.argv))
+        elif mode == CLI_MODE.SCRIPT:
+            with open(sys.argv[0], "r") as commands:
+                is_comment_block = False
+                for command in commands:
+                    if (
+                        command.startswith("#")
+                        or command.startswith("//")
+                        or command == "\n"
+                    ):
+                        continue
+                    elif command.startswith("/*"):
+                        is_comment_block = True
+                        continue
+                    elif command.startswith("*/"):
+                        is_comment_block = False
+                        continue
+                    elif is_comment_block:
+                        continue
+
+                    print("[green]> ", command)
+                    sliced_command = command.split()
+
+                    if sliced_command[0] == "VAR":
+                        var[sliced_command[1]] = sliced_command[2]
+                        continue
+
+                    sliced_command = parsing_variables(sliced_command)
+
+                    action(" ".join(sliced_command))
+        elif mode == CLI_MODE.REPL:
+            while True:
+                try:
+                    text = session.prompt(
+                        [("class:prompt", "❯ ")],  # ≻≻
+                        # cursor=CursorShape.BLINKING_BLOCK,
+                        # cursor=CursorShape.BLINKING_BEAM,
+                        cursor=CursorShape.BLINKING_UNDERLINE,
+                        bottom_toolbar=bottom_toolbar,
+                        complete_in_thread=True,
+                        complete_while_typing=True,
+                    )
+                    # remove whitespaces
+                    text = re.sub(r"^\s+", "", text)  # leading
+                    text = re.sub(r"\s+$", "", text)  # trailing
+                    text = re.sub(
+                        r"\s+", " ", text
+                    )  # replace multiple inline whitespaces with one
+                    if text in ["exit", "q", "quit"]:
+                        break
+                    else:
+                        if text == "":
+                            continue
+                        action(text)
+                except KeyboardInterrupt as ex:
+                    print(repr(ex))
+                    continue
+                except EOFError as _:
+                    print("[green]Exiting ...[/]")
+                    break
+            # else:
+            #    print('You entered:', repr(text))
+        print("[yellow]Good bye![/]")
+    except requests.exceptions.ConnectionError as _ : 
+        print("[yellow]Connection error[/]")

dmart/config.ini.sample

@@ -0,0 +1,7 @@
+url = "http://localhost:8282"
+shortname = "dmart"
+password = "xxxx"
+query_limit = 50
+retrieve_json_payload = True
+default_space = "management"
+pagination = 50

dmart/dmart.py

@@ -441,27 +441,26 @@ def main():
             
             # Try to find cli.py in the package or nearby
             try:
+                # When installed as a package, cli.py is in the same directory as dmart.py
+                dmart_dir = Path(__file__).resolve().parent
+                if str(dmart_dir) not in sys.path:
+                    sys.path.append(str(dmart_dir))
                 import cli
                 cli.main()
-            except ImportError:
-                # Check if it's in the same directory as dmart.py (installed via MANIFEST.in)
-                dmart_dir = Path(__file__).resolve().parent
-                cli_path = dmart_dir / "cli.py"
+            except (ImportError, AttributeError):
+                # Fallback for local development
+                cli_path = Path(__file__).resolve().parent.parent / "cli"
                 if cli_path.exists():
-                    if str(dmart_dir) not in sys.path:
-                        sys.path.append(str(dmart_dir))
-                    import cli
-                    cli.main()
-                else:
-                    # Fallback for local development
-                    cli_path = dmart_dir.parent / "cli"
-                    if cli_path.exists():
-                        sys.path.append(str(cli_path))
+                    sys.path.append(str(cli_path))
+                    try:
                         import cli
                         cli.main()
-                    else:
-                        print("Error: cli.py not found.")
+                    except (ImportError, AttributeError):
+                        print("Error: cli.py found but main() not found or import failed.")
                         sys.exit(1)
+                else:
+                    print("Error: cli.py not found.")
+                    sys.exit(1)
         case "serve":
             open_cxb = False
             if "--open-cxb" in sys.argv:

{dmart-1.4.39.dist-info → dmart-1.4.40.post1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dmart
-Version: 1.4.39
+Version: 1.4.40.post1
 Requires-Python: >=3.11
 Requires-Dist: fastapi
 Requires-Dist: pydantic

{dmart-1.4.39.dist-info → dmart-1.4.40.post1.dist-info}/RECORD

@@ -1,11 +1,13 @@
 dmart/__init__.py,sha256=xHodcATn-JYaoi3_TVDimR_UWG_ux9uLePUNQDzHhsY,122
 dmart/alembic.ini,sha256=wQweByyHQm-EI8BQkE0SHNRjULJ6Xn5jqgvv88IT5Sg,3738
 dmart/bundler.py,sha256=so8ZJResb1PcOH5vboa_mpFAdYr_T8u8DbbFXd570Lg,1704
+dmart/cli.py,sha256=aE3_2uHcEiQJA_f6viGRwUmGPB-OaNfzSgDgLR5gf40,40447
 dmart/config.env.sample,sha256=CKd4KIBeUatoFcO2IefmrVNBohpaVMQMFcoPkNBvCeI,696
+dmart/config.ini.sample,sha256=AADMdwG07zRV_cAGMoOorHQaqzTTL2WEIC_mPWjIslE,159
 dmart/conftest.py,sha256=0ry_zeCmdBNLbm5q115b-pkOrUFYxdsOUXbIMkr7E5Y,362
 dmart/curl.sh,sha256=lmHSFVr5ft-lc5Aq9LfvKyWfntrfYbnirhzx1EGjp_A,15743
 dmart/data_generator.py,sha256=CnE-VHEeX7-lAXtqCgbRqR9WHjTuOgeiZcviYrHAmho,2287
-dmart/dmart.py,sha256=Hmejqz79xFMJAJztnTEtqYm7XA1B1blIHryuZ48D8AI,24377
+dmart/dmart.py,sha256=Ax4GRara7_21T2Jcbl8EwGfdezr9NTz7UucW_oVw-pA,24419
 dmart/get_settings.py,sha256=Sbe2WCoiK398E7HY4SNLfDN_GmE8knR4M-YJWF31jcg,153
 dmart/hypercorn_config.toml,sha256=-eryppEG8HBOM_KbFc4dTQePnpyfoW9ZG5aUATU_6yM,50
 dmart/info.json,sha256=5Yz4tc5dEt5eVrM2LzLhXgN8vGvE0aDbbbEEaGDsD5k,123
@@ -278,8 +280,8 @@ dmart/utils/ticket_sys_utils.py,sha256=9QAlW2iiy8KyxQRBDj_WmzS5kKb0aYJmGwd4qzmGV
 dmart/utils/web_notifier.py,sha256=QM87VVid2grC5lK3NdS1yzz0z1wXljr4GChJOeK86W4,843
 dmart/utils/templates/activation.html.j2,sha256=XAMKCdoqONoc4ZQucD0yV-Pg5DlHHASZrTVItNS-iBE,640
 dmart/utils/templates/reminder.html.j2,sha256=aoS8bTs56q4hjAZKsb0jV9c-PIURBELuBOpT_qPZNVU,639
-dmart-1.4.39.dist-info/METADATA,sha256=GZGdPHieT1FqO4Baoab0MvxcAUm3DNXsnywyPMH6c08,2571
-dmart-1.4.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dmart-1.4.39.dist-info/entry_points.txt,sha256=N832M4wG8d2GDw1xztKRVM3TnxpY2QDzvdFE8XaWaG8,43
-dmart-1.4.39.dist-info/top_level.txt,sha256=zJo4qk9fUW0BGIR9f4DCfpxaXbvQXH9izIOom6JsyAo,6
-dmart-1.4.39.dist-info/RECORD,,
+dmart-1.4.40.post1.dist-info/METADATA,sha256=dRF5fMCJ7hyZsIKXGG6o5e5H00JxS8JkqucCNR4aINc,2577
+dmart-1.4.40.post1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dmart-1.4.40.post1.dist-info/entry_points.txt,sha256=N832M4wG8d2GDw1xztKRVM3TnxpY2QDzvdFE8XaWaG8,43
+dmart-1.4.40.post1.dist-info/top_level.txt,sha256=zJo4qk9fUW0BGIR9f4DCfpxaXbvQXH9izIOom6JsyAo,6
+dmart-1.4.40.post1.dist-info/RECORD,,