mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-23 09:24:31 +02:00
8d74ffc932* Add comprehensive type annotations - mypy.ini: expanded configuration with module-specific settings - hookspecs.py: type annotations for hook functions - plugins.py: typed get_plugins() return value - recipes.py: full type annotations for parsedate, parsedatetime, jsonsplit - utils.py: extensive type annotations including Row type alias, TypeTracker, ValueTracker, and all utility functions - db.py: type annotations for Database methods (__exit__, ensure_autocommit_off, tracer, register_function, etc.) and Queryable class methods - tests/test_docs.py: updated to match new signature display format * Fix type errors caught by ty check - Add type: ignore comments for external library type stub limitations (csv.reader, click.progressbar, IOBase.name, Callable.__name__) - Change Iterable to Sequence for SQL where_args parameters - Use db.table() instead of db[name] for proper Table return type - Fix rebuild_fts return type from None to Table - Update test_tracer to expect fewer queries (optimization side effect) * Fix mypy type errors - Add type: ignore comments for runtime-valid patterns mypy can't verify - Fix new_column_types annotation to Dict[str, Set[type]] - Add type: ignore for Default sentinel values passed to create_table * mypy skip tests directory * Fix CI: exclude typing imports from recipe docs, skip mypy on tests - Add Callable and Optional to exclusion list in _generate_convert_help() - Regenerate docs/cli-reference.rst with cog - Add [mypy-tests.*] ignore_errors = True to skip test type errors --------- Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
76 lines
2 KiB
Python
76 lines
2 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Callable, Optional
|
|
|
|
from dateutil import parser
|
|
import json
|
|
|
|
IGNORE: object = object()
|
|
SET_NULL: object = object()
|
|
|
|
|
|
def parsedate(
|
|
value: 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: 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], object] = 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)])
|