remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-24 09:54:31 +02:00
Code Issues Projects Releases Packages Wiki Activity

Added support for bytes and datetime.datetime

Browse source
This commit is contained in:
Simon Willison 2019-01-24 18:59:21 -08:00
commit 6ad9037c96
2 changed files with 25 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

21
sqlite_utils/db.py
View file

@ -1,5 +1,6 @@
import sqlite3
from collections import namedtuple
import datetime
import json
Column = namedtuple(
@ -53,16 +54,20 @@ class Database:
key=lambda p: column_order.index(p[0]) if p[0] in column_order else 999
)
extra = ""
col_type_mapping = {
float: "FLOAT",
int: "INTEGER",
bool: "INTEGER",
str: "TEXT",
bytes.__class__: "BLOB",
bytes: "BLOB",
datetime.datetime: "TEXT",
None.__class__: "TEXT",
}
columns_sql = ",\n".join(
" [{col_name}] {col_type} {primary_key} {references}".format(
col_name=col_name,
col_type={
float: "FLOAT",
int: "INTEGER",
bool: "INTEGER",
str: "TEXT",
None.__class__: "TEXT",
}[col_type],
col_type=col_type_mapping[col_type],
primary_key=" PRIMARY KEY" if (pk == col_name) else "",
references=(
" REFERENCES [{other_table}]([{other_column}])".format(
@ -252,6 +257,8 @@ class Table:
t = int
elif {int, float, bool}.issuperset(types):
t = float
elif {bytes, str}.issuperset(types):
t = bytes
else:
t = str
column_types[key] = t