diff options
Diffstat (limited to 'girok/api/category.py')
| -rw-r--r-- | girok/api/category.py | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/girok/api/category.py b/girok/api/category.py new file mode 100644 index 0000000..f01d274 --- /dev/null +++ b/girok/api/category.py @@ -0,0 +1,131 @@ +import requests +from rich.console import Console + +from girok.config import get_config +import girok.utils.auth as auth_utils +import girok.utils.general as general_utils +import girok.utils.display as display_utils +import girok.constants as constants + +console = Console() +cfg = get_config() + +def get_categories(): + resp = requests.get( + cfg.base_url + "/categories", + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + if resp.status_code == 200: + return general_utils.bytes2dict(resp.content) + + + +def add_category(cat_str: str, color=None): + cats = cat_str.split('/') + resp = requests.post( + cfg.base_url + "/categories", + json={ + "names": cats, + "color": color + }, + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + return resp + + +def remove_category(cat_str: str): + cats = cat_str.split('/') + resp = requests.delete( + cfg.base_url + "/categories", + json={ + "cats": cats + }, + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + + return resp + + +def rename_category(cat_str: str, new_name: str): + cats = cat_str.split('/') + resp = requests.patch( + cfg.base_url + "/categories/name", + json={ + "cats": cats, + "new_name": new_name + }, + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + + return resp + + +def move_category(cat_str: str, new_parent_cat_str: str): + if cat_str.endswith('/'): + cat_str = cat_str[:-1] + if new_parent_cat_str.endswith('/'): + new_parent_cat_str = new_parent_cat_str[:-1] + + cats = cat_str.split('/') if cat_str else [] + new_parent_cats = new_parent_cat_str.split('/') if new_parent_cat_str else [] + resp = requests.patch( + cfg.base_url + "/categories/parent", + json={ + "cats": cats, + "new_parent_cats": new_parent_cats + }, + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + return resp + + +def get_last_cat_id(cats: list): + resp = requests.get( + cfg.base_url + "/categories/last-cat-id", + json={ + "cats": cats + }, + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + if resp.status_code == 200: + return general_utils.bytes2dict(resp.content)['cat_id'] + elif resp.status_code == 400: + err_msg = general_utils.bytes2dict(resp.content)['detail'] + display_utils.center_print(err_msg, constants.DISPLAY_TERMINAL_COLOR_ERROR) + exit(0) + else: + display_utils.center_print(resp.content, constants.DISPLAY_TERMINAL_COLOR_ERROR) + exit(0) + + +def get_category_color(cat_id: int): + resp = requests.get( + cfg.base_url + f"/categories/{cat_id}/color", + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + + if resp.status_code == 200: + return general_utils.bytes2dict(resp.content)['color'] + elif resp.status_code == 400: + err_msg = general_utils.bytes2dict(resp.content)['detail'] + display_utils.center_print(err_msg, constants.DISPLAY_TERMINAL_COLOR_ERROR) + exit(0) + else: + display_utils.center_print(resp.content, constants.DISPLAY_TERMINAL_COLOR_ERROR) + exit(0) + + +def get_color_dict(): + resp = requests.get( + cfg.base_url + "/categories/color", + headers=auth_utils.build_jwt_header(cfg.config_path) + ) + if resp.status_code == 200: + color_dict = general_utils.bytes2dict(resp.content) + return color_dict + elif resp.status_code == 400: + err_msg = general_utils.bytes2dict(resp.content)['detail'] + display_utils.center_print(err_msg, constants.DISPLAY_TERMINAL_COLOR_ERROR) + else: + display_utils.center_print("Error occurred.", constants.DISPLAY_TERMINAL_COLOR_ERROR) +
\ No newline at end of file |