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

subclass namedtuple for foreign key

Browse source
This commit is contained in:
David Kane 2020-11-16 10:50:09 +00:00
commit 99992deb31
1 changed files with 16 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

58
sqlite_utils/db.py
View file

@ -51,22 +51,26 @@ except ImportError:
Column = namedtuple(
"Column", ("cid", "name", "type", "notnull", "default_value", "is_pk")
)
ForeignKeyBase = namedtuple(
"ForeignKeyBase", ("table", "column", "other_table", "other_column")
)
Index = namedtuple("Index", ("seq", "name", "unique", "origin", "partial", "columns"))
Trigger = namedtuple("Trigger", ("name", "table", "sql"))
class ForeignKey:
def __init__(self, table, column, other_table, other_column):
self.table = table
if isinstance(column, str):
self.column = (column,)
elif isinstance(column, (tuple, list)):
self.column = tuple(column)
self.other_table = other_table
if isinstance(other_column, str):
self.other_column = (other_column,)
elif isinstance(other_column, (tuple, list)):
self.other_column = tuple(other_column)
class ForeignKey(ForeignKeyBase):
def __new__(cls, table, column, other_table, other_column):
# column and other_column should be a tuple
if isinstance(column, (tuple, list)):
column = tuple(column)
else:
column = (column,)
if isinstance(other_column, (tuple, list)):
other_column = tuple(other_column)
else:
other_column = (other_column,)
self = super(ForeignKey, cls).__new__(cls, table, column, other_table, other_column)
return self
@property
def column_str(self):
@ -76,36 +80,6 @@ class ForeignKey:
def other_column_str(self):
return ",".join(["[{}]".format(c) for c in self.other_column])
def __eq__(self, other):
if isinstance(other, ForeignKey):
return all(
(
self.table == other.table,
self.column == other.column,
self.other_table == other.other_table,
self.other_column == other.other_column,
)
)
return False
def __lt__(self, other):
if isinstance(other, ForeignKey):
return (self.table, self.column, self.other_table, self.other_column) < (
other.table,
other.column,
other.other_table,
other.other_column,
)
return False
def __repr__(self):
return "ForeignKey({table}({column}), {other_table}({other_column}))".format(
table=self.table,
column=self.column_str,
other_table=self.other_table,
other_column=self.other_column_str,
)
@property
def sql(self):
return (