From 35039880ba9a9b7637ea4c4c0a66f63c2472a1bf Mon Sep 17 00:00:00 2001 From: Eugene Crosser Date: Sat, 28 May 2022 20:07:30 +0200 Subject: [PATCH] typeckecing: annotate ocid_dload.py --- gps303/ocid_dload.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/gps303/ocid_dload.py b/gps303/ocid_dload.py index 0bdb8b7..43f5893 100644 --- a/gps303/ocid_dload.py +++ b/gps303/ocid_dload.py @@ -1,9 +1,10 @@ -from configparser import NoOptionError +from configparser import ConfigParser, NoOptionError import csv from logging import getLogger import requests from sqlite3 import connect -from zlib import decompressobj, MAX_WBITS +from typing import IO, Optional +from zlib import decompressobj, MAX_WBITS, _Decompress from . import common @@ -41,13 +42,13 @@ class unzipped: and yelds them as strings. """ - def __init__(self, zstream): + def __init__(self, zstream: IO[bytes]) -> None: self.zstream = zstream - self.decoder = decompressobj(16 + MAX_WBITS) + self.decoder: Optional[_Decompress] = decompressobj(16 + MAX_WBITS) self.outdata = b"" self.line = b"" - def read(self, n=None): + def read(self, n: int = 1024) -> bytes: if self.decoder is None: return b"" while len(self.outdata) < n: @@ -61,7 +62,7 @@ class unzipped: return data return b"" - def __next__(self): + def __next__(self) -> str: while True: splittry = self.line.split(b"\n", maxsplit=1) if len(splittry) > 1: @@ -74,11 +75,11 @@ class unzipped: self.line = rest return line.decode("utf-8") - def __iter__(self): + def __iter__(self) -> "unzipped": return self -def main(conf): +def main(conf: ConfigParser) -> None: try: url = conf.get("opencellid", "downloadurl") mcc = "" -- 2.39.2