mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-29 12:24:32 +02:00
83c50527aa- Improve mypy configuration with appropriate strictness settings
- Add comprehensive type hints to public API functions:
- hookspecs.py: Type register_commands and prepare_connection hooks
- plugins.py: Type get_plugins function and PluginManager
- recipes.py: Type parsedate, parsedatetime, jsonsplit functions
- utils.py: Type all public utility functions including rows_from_file,
TypeTracker, ValueTracker, chunks, hash_record, flatten, etc.
- db.py: Type Database, Queryable, Table, and View class methods
including constructor, execute, query, table operations, etc.
- Exclude cli.py from strict checking (internal implementation detail)
- All public API modules now pass mypy type checking
74 lines
2 KiB
Python
74 lines
2 KiB
Python
from typing import Any, Callable, Optional
|
|
|
|
from dateutil import parser
|
|
import json
|
|
|
|
IGNORE: object = object()
|
|
SET_NULL: object = object()
|
|
|
|
|
|
def parsedate(
|
|
value: Optional[str],
|
|
dayfirst: bool = False,
|
|
yearfirst: bool = False,
|
|
errors: Optional[object] = None,
|
|
) -> Optional[str]:
|
|
"""
|
|
Parse a date and convert it to ISO date format: yyyy-mm-dd
|
|
\b
|
|
- dayfirst=True: treat xx as the day in xx/yy/zz
|
|
- yearfirst=True: treat xx as the year in xx/yy/zz
|
|
- errors=r.IGNORE to ignore values that cannot be parsed
|
|
- errors=r.SET_NULL to set values that cannot be parsed to null
|
|
"""
|
|
if not value:
|
|
return value
|
|
try:
|
|
return (
|
|
parser.parse(value, dayfirst=dayfirst, yearfirst=yearfirst)
|
|
.date()
|
|
.isoformat()
|
|
)
|
|
except parser.ParserError:
|
|
if errors is IGNORE:
|
|
return value
|
|
elif errors is SET_NULL:
|
|
return None
|
|
else:
|
|
raise
|
|
|
|
|
|
def parsedatetime(
|
|
value: Optional[str],
|
|
dayfirst: bool = False,
|
|
yearfirst: bool = False,
|
|
errors: Optional[object] = None,
|
|
) -> Optional[str]:
|
|
"""
|
|
Parse a datetime and convert it to ISO datetime format: yyyy-mm-ddTHH:MM:SS
|
|
\b
|
|
- dayfirst=True: treat xx as the day in xx/yy/zz
|
|
- yearfirst=True: treat xx as the year in xx/yy/zz
|
|
- errors=r.IGNORE to ignore values that cannot be parsed
|
|
- errors=r.SET_NULL to set values that cannot be parsed to null
|
|
"""
|
|
if not value:
|
|
return value
|
|
try:
|
|
return parser.parse(value, dayfirst=dayfirst, yearfirst=yearfirst).isoformat()
|
|
except parser.ParserError:
|
|
if errors is IGNORE:
|
|
return value
|
|
elif errors is SET_NULL:
|
|
return None
|
|
else:
|
|
raise
|
|
|
|
|
|
def jsonsplit(
|
|
value: str, delimiter: str = ",", type: Callable[[str], Any] = str
|
|
) -> str:
|
|
"""
|
|
Convert a string like a,b,c into a JSON array ["a", "b", "c"]
|
|
"""
|
|
return json.dumps([type(s.strip()) for s in value.split(delimiter)])
|