winzer/elwig-misc
3
1
Fork 0
Code Pull Requests Activity

Update REGEXP function for sqlite

Browse Source
This commit is contained in:
Lorenz Stechauner
2023-08-18 15:33:09 +02:00
parent 5d92f909a1
commit af94afaa62
2 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
wgmaster/import.py
View File

@ -1,11 +1,10 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import List, Optional
from typing import List
import argparse
import sqlite3
import os
import re
import datetime
import utils
@ -39,12 +38,6 @@ def get_sql_files() -> List[str]:
return files
def sqlite_regexp(pattern: str, value: Optional[str]) -> Optional[bool]:
if value is None:
return None
return re.match(pattern, value) is not None
def import_csv(cur: sqlite3.Cursor, table_name: str, verbose: bool = False) -> None:
rows = utils.csv_parse(f'{DIR}/{table_name}.csv')
names = next(rows)
@ -117,7 +110,7 @@ def main() -> None:
sqlite3.register_adapter(datetime.time, lambda t: t.strftime('%H:%M:%S'))
cnx = sqlite3.connect(args.db)
cnx.create_function('REGEXP', 2, sqlite_regexp)
cnx.create_function('REGEXP', 2, utils.sqlite_regexp, deterministic=True)
if not args.keep:
for file_name in get_sql_files():