Skip to content

Commit

Permalink
minor fixes to make insertion of identifiers work
Browse files Browse the repository at this point in the history
  • Loading branch information
for-hyde committed Nov 14, 2024
1 parent 0787555 commit 9869749
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
13 changes: 6 additions & 7 deletions omnipath_metabo/schema/_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from ._base import Base
from ._connection import Connection
from ..misc._tee import Tee
from .. import data as _data

_log("Hello from main module")

Expand Down Expand Up @@ -129,25 +128,25 @@ def load(self):

self.create()

resource_labels = self.resource.id_types
resource_labels = [self.resource.name] + self.resource.id_types

insert_resource = (
insert(_structure.Resource).
values(name = resource_labels).
returning(_structure.Resource.id).
on_conflict_do_update(
returning(_structure.Resource.id))

insert_resource = insert_resource.on_conflict_do_update(
index_elements=['name'],
set_ = {
'name':insert_resource.excluded.name
}
)
)

resids = self.session.execute(insert_resource).fetchall()
self.session.commit()
resource_key = {
label: _id[0]
for label, _id in zip(resource_labels, resid)
for label, _id in zip(resource_labels, resids)
}

_log(f'loading resource {self.resource.name}', level = -1)
Expand Down Expand Up @@ -189,7 +188,7 @@ def load(self):
_log('resource ids collected.')

insert_ids = (
(_id, strids[smiles], resource_key, i==0, self._id_type(resource_key, i-1))
(_id, strids[smiles], resource_key, i==0, resource_labels[i])
for name, smiles, _ in (
r['structure'] for r in cached_resource.cached['struct']
)
Expand Down
1 change: 1 addition & 0 deletions omnipath_metabo/schema/_structure.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from sqlalchemy.orm import relationship
from ._base import Base
from pypath.inputs import hmdb, swisslipids, lipidmaps, ramp
from .. import data as _data

class MolType(types.UserDefinedType):
cache_ok = True
Expand Down

0 comments on commit 9869749

Please sign in to comment.