From: Javier Sancho Date: Thu, 24 May 2018 14:56:09 +0000 (+0200) Subject: PyTables Row class based in list class for allow json serialization X-Git-Url: https://git.jsancho.org/?a=commitdiff_plain;h=48742e0185fe1d0c1babe0db188cffe284d6ce88;p=datasette-pytables.git PyTables Row class based in list class for allow json serialization --- diff --git a/datasette_pytables/__init__.py b/datasette_pytables/__init__.py index 5d9fb22..d9530ae 100644 --- a/datasette_pytables/__init__.py +++ b/datasette_pytables/__init__.py @@ -232,12 +232,29 @@ class Connection: else: raise Exception("SQLite queries cannot be executed with this connector") -class Row(OrderedDict): - def __getitem__(self, label): - if type(label) is int: - return super(OrderedDict, self).__getitem__(list(self.keys())[label]) +class Row(list): + def __init__(self, values=None): + self.labels = [] + self.values = [] + if values: + for idx in values: + self.__setitem__(idx, values[idx]) + + def __setitem__(self, idx, value): + if type(idx) is str: + if idx in self.labels: + self.values[self.labels.index(idx)] = value + else: + self.labels.append(idx) + self.values.append(value) + else: + self.values[idx] = value + + def __getitem__(self, idx): + if type(idx) is str: + return self.values[self.labels.index(idx)] else: - return super(OrderedDict, self).__getitem__(label) + return self.values[idx] def __iter__(self): - return self.values().__iter__() + return self.values.__iter__()