Fix typing

- move info to __init__.py
- delete info.py
- Union to Optional
This commit is contained in:
xSTACIA 2022-08-13 13:35:56 +07:00
parent f10e8b5432
commit 3547662567
7 changed files with 16 additions and 21 deletions

View File

@ -1,11 +1,10 @@
VERSION = "1.2.10dev0"
AUTHOR = "M-307"
from .client import *
from .exception import *
from .model import *
from .utils import *
from .info import *
from .cache import *
from .enum import *
from .assets import *
__VERSION__ = VERSION
__AUTHOR__ = AUTHOR
from .assets import *

View File

@ -2,13 +2,11 @@ import json
import os
import logging
from typing import Dict, List, TextIO
from .enum import Language
from .model import assets
from .utils import create_ui_path
from typing import Union
from typing import Dict, List, TextIO, Optional, Union
PATH = os.path.dirname(os.path.abspath(__file__))
@ -41,7 +39,7 @@ class Assets:
return [x for x in self.DATA["costumes"]]
@classmethod
def character(cls, id: Union[int, str]) -> Union[assets.CharacterAsset, None]: # noqa: E501
def character(cls, id: Union[int, str]) -> Optional[assets.CharacterAsset]: # noqa: E501
LOGGER.debug(f"Getting character assets with id: {id}")
data = cls.DATA["characters"].get(str(id))
@ -57,7 +55,7 @@ class Assets:
})
@classmethod
def character_costume(cls, id: int): # noqa: E501
def character_costume(cls, id: int):
LOGGER.debug(f"Getting costume assets with id: {id}")
data = cls.DATA["costumes"].get(str(id))
if not data:
@ -70,7 +68,7 @@ class Assets:
})
@classmethod
def constellations(cls, id: int) -> Union[assets.CharacterConstellationsAsset, None]: # noqa: E501
def constellations(cls, id: int) -> Optional[assets.CharacterConstellationsAsset]:
LOGGER.debug(f"Getting character constellations assets with id: {id}")
data = cls.DATA["constellations"].get(str(id))
if not data:
@ -84,7 +82,7 @@ class Assets:
})
@classmethod
def skills(cls, id: int) -> Union[assets.CharacterSkillAsset, None]:
def skills(cls, id: int) -> Optional[assets.CharacterSkillAsset]:
LOGGER.debug(f"Getting character skills assets with id: {id}")
data = cls.DATA["skills"].get(str(id))
@ -99,7 +97,7 @@ class Assets:
})
@classmethod
def namecards(cls, id: int) -> Union[assets.NamecardAsset, None]:
def namecards(cls, id: int) -> Optional[assets.NamecardAsset]:
LOGGER.debug(f"Getting namecards assets with id: {id}")
data = cls.DATA["namecards"].get(str(id))
if not data:
@ -115,7 +113,7 @@ class Assets:
})
@classmethod
def get_hash_map(cls, hash_id: str) -> Union[str, None]:
def get_hash_map(cls, hash_id: str) -> Optional[str]:
LOGGER.debug(f"Getting nameTextMapHash {hash_id} with language: {cls.LANGS}") # noqa: E501
for key in cls.HASH_MAP:
if str(hash_id) in cls.HASH_MAP[key]:
@ -127,7 +125,7 @@ class Assets:
return
@classmethod
def character_icon(cls, id: int) -> Union[assets.CharacterIconAsset, None]:
def character_icon(cls, id: int) -> Optional[assets.CharacterIconAsset]:
data = cls.character(id)
if not data:
return

View File

@ -73,7 +73,7 @@ class EnkaNetworkAPI:
self.LOGGER.debug(f"Validating with UID {uid}...")
if self._enable_cache:
self.LOGGER.warn("Getting data from cache...")
self.LOGGER.warning("Getting data from cache...")
data = await self.cache.get(uid)
if data is not None:

View File

@ -85,7 +85,7 @@ class HTTPClient:
kwargs['headers'] = {**utils.get_default_header(), **self.__headers}
response: Optional[aiohttp.ClientResponse] = None
data: Optional[Union[Dict[str, Any]]] = None
data: Optional[Union[Dict[str, Any], str]] = None
if self.__session is MISSING:
self.__session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=self.__timeout))

View File

@ -1,2 +0,0 @@
VERSION = "1.2.10dev0"
AUTHOR = "M-307"

View File

@ -7,7 +7,7 @@ import sys
from typing import Any, Dict, TYPE_CHECKING
from .info import VERSION
from . import VERSION
if TYPE_CHECKING:
from aiohttp import ClientResponse

View File

@ -1,7 +1,7 @@
import setuptools
import re
with open('enkanetwork/info.py') as f:
with open('enkanetwork/__init__.py') as f:
"""
Get version from utils.py
Ref: https://github.com/Rapptz/discord.py/blob/52f3a3496bea13fefc08b38f9ed01641e565d0eb/setup.py#L9