dm1sh/lenengro_parser
1
0
Fork 0
Code Pull Requests Activity

Sorted parser lib exports, fixed preprocess types

Browse Source
This commit is contained in:
Dmitriy Shishkov 2023-10-11 23:13:40 +03:00
parent 9cd21f717d
commit f3d24fe701
Signed by: dm1sh
GPG Key ID: 027994B0AA357688
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
parser/__init__.py
View File

@ -1,10 +1,12 @@
from .address import split_addresses from .address import split_addresses
from .building_id import ( from .building_id import (
async_fetch_building_id,
async_fetch_building_ids, async_fetch_building_ids,
concurrent_fetch_builing_ids, concurrent_fetch_builing_ids,
fetch_builing_ids, fetch_builing_ids,
get_building_id, get_building_id,
) )
from .lenenergo import LenenergoParser
from .preprocess import ( from .preprocess import (
COL_NS, COL_NS,
ICOL_NS, ICOL_NS,
@ -12,20 +14,20 @@ from .preprocess import (
preprocess_df, preprocess_df,
preprocess_read_df, preprocess_read_df,
) )
from .lenenergo import LenenergoParser
from .util import pipeline from .util import pipeline
__all__ = ( __all__ = (
"LenenergoParser", "async_fetch_building_id",
"split_addresses",
"get_building_id",
"fetch_builing_ids",
"async_fetch_building_ids", "async_fetch_building_ids",
"concurrent_fetch_builing_ids",
"preprocess_df",
"COL_NS", "COL_NS",
"ICOL_NS", "concurrent_fetch_builing_ids",
"preprocess_read_df", "fetch_builing_ids",
"get_building_id",
"group_by_index", "group_by_index",
"ICOL_NS",
"LenenergoParser",
"pipeline", "pipeline",
"preprocess_df",
"preprocess_read_df",
"split_addresses",
) )

2
parser/preprocess.py
View File

@ -34,6 +34,8 @@ def preprocess_df(df: pd.DataFrame) -> pd.DataFrame:
) )
df.drop(columns=[f"{a}_date", f"{a}_time"], inplace=True) df.drop(columns=[f"{a}_date", f"{a}_time"], inplace=True)
df = df.convert_dtypes()
return df return df