mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-26 19:04:32 +02:00
69a1c0d960* Automated upgrades by Ruff
uvx --with 'ruff>=0.16.0' ruff check . --fix --unsafe-fixes
* Fix remaining Ruff errors with GPT-5.6 Sol high
https://gist.github.com/simonw/6da7906a9fea6e90da131c21a9055199
* Fix flake E501 long lines
* New Protocol for migrations to make ty happy
76 lines
2 KiB
Python
76 lines
2 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from collections.abc import Callable
|
|
|
|
from dateutil import parser
|
|
|
|
IGNORE: object = object()
|
|
SET_NULL: object = object()
|
|
|
|
|
|
def parsedate(
|
|
value: str,
|
|
dayfirst: bool = False,
|
|
yearfirst: bool = False,
|
|
errors: object | None = None,
|
|
) -> str | None:
|
|
"""
|
|
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: object | None = None,
|
|
) -> str | None:
|
|
"""
|
|
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)])
|