Skip to content
This repository was archived by the owner on Dec 23, 2024. It is now read-only.

Fix type hints #1289

Open
mo1ein wants to merge 6 commits into
base: master
Choose a base branch
from
Open

Fix type hints #1289

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
79ca38c
Add type hint in parser
mo1ein Jun 22, 2023
3cf6aac
Add type hint in storage
mo1ein Jun 22, 2023
c628b21
Add type hint in errors
mo1ein Jun 22, 2023
fa95030
Add type hint in handlers
mo1ein Jun 22, 2023
d694c70
Add type hints in compiler
mo1ein Jun 22, 2023
90b9dd2
Add type hints in raw/core
mo1ein Jun 23, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add type hint in storage
  • Loading branch information
@mo1ein
mo1ein committed Jun 22, 2023
commit 3cf6aac9996001f8b0be9d3acff4a5452fbec05f
8 changes: 4 additions & 4 deletions pyrogram/storage/file_storage.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,12 +29,12 @@
class FileStorage(SQLiteStorage):
FILE_EXTENSION = ".session"

def __init__(self, name: str, workdir: Path):
def __init__(self, name: str, workdir: Path) -> None:
super().__init__(name)

self.database = workdir / (self.name + self.FILE_EXTENSION)

def update(self):
def update(self) -> None:
version = self.version()

if version == 1:
Expand All @@ -51,7 +51,7 @@ def update(self):

self.version(version)

async def open(self):
async def open(self) -> None:
path = self.database
file_exists = path.is_file()

Expand All @@ -65,5 +65,5 @@ async def open(self):
with self.conn:
self.conn.execute("VACUUM")

async def delete(self):
async def delete(self) -> None:
os.remove(self.database)
6 changes: 3 additions & 3 deletions pyrogram/storage/memory_storage.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,12 +27,12 @@


class MemoryStorage(SQLiteStorage):
def __init__(self, name: str, session_string: str = None):
def __init__(self, name: str, session_string: str = None) -> None:
super().__init__(name)

self.session_string = session_string

async def open(self):
async def open(self) -> None:
self.conn = sqlite3.connect(":memory:", check_same_thread=False)
self.create()

Expand Down Expand Up @@ -69,5 +69,5 @@ async def open(self):
await self.is_bot(is_bot)
await self.date(0)

async def delete(self):
async def delete(self) -> None:
pass
34 changes: 17 additions & 17 deletions pyrogram/storage/storage.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,55 +29,55 @@ class Storage:

SESSION_STRING_FORMAT = ">BI?256sQ?"

def __init__(self, name: str):
def __init__(self, name: str) -> None:
self.name = name

async def open(self):
async def open(self) -> None:
raise NotImplementedError

async def save(self):
async def save(self) -> None:
raise NotImplementedError

async def close(self):
async def close(self) -> None:
raise NotImplementedError

async def delete(self):
async def delete(self) -> None:
raise NotImplementedError

async def update_peers(self, peers: List[Tuple[int, int, str, str, str]]):
async def update_peers(self, peers: List[Tuple[int, int, str, str, str]]) -> None:
raise NotImplementedError

async def get_peer_by_id(self, peer_id: int):
async def get_peer_by_id(self, peer_id: int) -> None:
raise NotImplementedError

async def get_peer_by_username(self, username: str):
async def get_peer_by_username(self, username: str) -> None:
raise NotImplementedError

async def get_peer_by_phone_number(self, phone_number: str):
async def get_peer_by_phone_number(self, phone_number: str) -> None:
raise NotImplementedError

async def dc_id(self, value: int = object):
async def dc_id(self, value: int = object) -> None:
raise NotImplementedError

async def api_id(self, value: int = object):
async def api_id(self, value: int = object) -> None:
raise NotImplementedError

async def test_mode(self, value: bool = object):
async def test_mode(self, value: bool = object) -> None:
raise NotImplementedError

async def auth_key(self, value: bytes = object):
async def auth_key(self, value: bytes = object) -> None:
raise NotImplementedError

async def date(self, value: int = object):
async def date(self, value: int = object) -> None:
raise NotImplementedError

async def user_id(self, value: int = object):
async def user_id(self, value: int = object) -> None:
raise NotImplementedError

async def is_bot(self, value: bool = object):
async def is_bot(self, value: bool = object) -> None:
raise NotImplementedError

async def export_session_string(self):
async def export_session_string(self) -> str:
packed = struct.pack(
self.SESSION_STRING_FORMAT,
await self.dc_id(),
Expand Down