tracking_policy_agendas.api

Complexity

Total Complexity

4

Size/Duplication

Total Lines	62
Duplicated Lines	19.35 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	downloader()	0	12	2
A	get_resources()	12	12	2

"""
API
....................................................................................................
MIT License
Copyright (c) 2021-2023 AUT Iran, Mohammad H Forouhesh
Copyright (c) 2021-2022 MetoData.ai, Mohammad H Forouhesh
....................................................................................................
This module contains tools to download resources over http connections.
supported http links are:
    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/xgb_vaccine.zip'

This line is too long as per the coding-style (109/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/pa_vaccine.zip'

This line is too long as per the coding-style (108/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/lasso_vaccine.zip'

This line is too long as per the coding-style (111/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/gnb_vaccine.zip'

This line is too long as per the coding-style (109/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/xgb_jcpoa.zip'

This line is too long as per the coding-style (107/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/pa_jcpoa.zip'

This line is too long as per the coding-style (106/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/lasso_jcpoa.zip',

This line is too long as per the coding-style (110/100).

    - 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/gnb_jcpoa.zip'

This line is too long as per the coding-style (107/100).

"""

import os
from typing import Union
import zipfile
from io import BytesIO
import requests

http_dict = {'xgb_vaccine': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/bin/xgb_vaccine.zip',

This line is too long as per the coding-style (129/100).

             'pa_vaccine': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/bin/pa_vaccine.zip',

This line is too long as per the coding-style (127/100).

             'lasso_vaccine': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/bin/lasso_vaccine.zip',

This line is too long as per the coding-style (133/100).

             'gnb_vaccine': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/bin/gnb_vaccine.zip',

This line is too long as per the coding-style (129/100).

             'xgb_jcpoa': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/xgb_jcpoa.zip',

This line is too long as per the coding-style (128/100).

             'pa_jcpoa': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/pa_jcpoa.zip',

This line is too long as per the coding-style (126/100).

             'lasso_jcpoa': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/lasso_jcpoa.zip',

This line is too long as per the coding-style (132/100).

             'gnb_jcpoa': 'https://github.com/MohammadForouhesh/tracking-policy-agendas/releases/download/v1.0.0/gnb_jcpoa.zip'}

This line is too long as per the coding-style (128/100).

def downloader(path: str, save_path: str) -> Union[int, None]:
    """
    A tool to download and save files over https.
    :param path:        The path to the desired file.
    :param save_path:   The intended storage path.
    :return:            If the file exists, it returns 0 (int), otherwise nothing would be returned.
    """
    assert path in http_dict.values(), f'{path[-path[::-1].find("/"):]} is not a supported models, use: \n{http_dict}'

This line is too long as per the coding-style (118/100).

    model_bin = requests.get(path, allow_redirects=True)
    with zipfile.ZipFile(BytesIO(model_bin.content)) as resource:
        resource.extractall(save_path)
    return 0


def get_resources(dir_path: str, resource_name: str) -> Union[int, str]:

This code seems to be duplicated in your project.

    """
    A tool to download required resources over internet.
    :param dir_path:        Path to the https link of the resource
    :param resource_name:   Resource name.
    :return:                Path to the downloaded resource.
    """
    save_dir = dir_path + 'model_dir/'
    if os.path.isdir(save_dir + f'{resource_name}/'): return 0

More than one statement on a single line

    os.makedirs(save_dir, exist_ok=True)
    downloader(path=http_dict[resource_name], save_path=save_dir)
    return str(save_dir + resource_name)