Skip to content

Commit ac09fec

Browse files
author
sreeder
committed
add delete values
1 parent 860a47d commit ac09fec

3 files changed

Lines changed: 14 additions & 5 deletions

File tree

odm2api/ODM2/models.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2050,21 +2050,22 @@ def _changeSchema(schema):
20502050

20512051
if isinstance(Tbl, api.DeclarativeMeta):
20522052
Tbl.__table__.schema = schema
2053+
Tbl.__table_args__["schema"]=schema
20532054

20542055

20552056
def _getSchema(engine):
20562057
from sqlalchemy.engine import reflection
20572058

20582059
insp = reflection.Inspector.from_engine(engine)
2059-
20602060
for name in insp.get_schema_names():
20612061
if 'odm2' == name.lower():
20622062
return name
2063-
else:
2064-
return insp.default_schema_name
20652063

2064+
return insp.default_schema_name
20662065

20672066
def setSchema(engine):
20682067
s = _getSchema(engine)
2068+
if s is None:
2069+
s = ''
20692070
_changeSchema(s)
20702071

odm2api/ODM2/services/deleteService.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,14 @@ def test(self):
9090

9191

9292
# ################################################################################
93-
# ODM2
93+
# Result Values
9494
# ################################################################################
9595

96+
def deleteTSRValues(self, ids=None, dates=None):
97+
98+
q = self._session.query(TimeSeriesResultValues)
99+
if ids:
100+
q = q.filter(TimeSeriesResultValues.ResultID.in_(ids))
101+
if dates:
102+
q = q.filter(TimeSeriesResultValues.ValueDateTime.in_(dates))
103+
return q.delete(False)

odm2api/ODMconnection.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def __init__(self, connection_string, echo=True, version = 2.0):
3030
# Create session maker
3131
self.Session = sessionmaker(bind=self.engine)
3232
self.test_Session = sessionmaker(bind=self.test_engine)
33+
setSchema(self.engine)
3334
self.version=version
3435

3536
def getSession(self):
@@ -65,7 +66,6 @@ def createConnection(self, engine, address, db=None, user=None, password=None, d
6566
@classmethod
6667
def createConnectionFromString(self, conn_string, dbtype= 2.0, echo = False):
6768
s = SessionFactory(conn_string, echo=echo, version=dbtype)
68-
setSchema(s.engine)
6969
return s
7070

7171
@classmethod

0 commit comments

Comments
 (0)