@@ -234,7 +234,7 @@ def restore(self):
234234 self ._script ("edit_service.restore()\n " , 'black' )
235235 Publisher .sendMessage ("scroll" )
236236
237- def saveFactory (self , var = None , method = None , qcl = None ):
237+ def saveFactory (self , var = None , method = None , qcl = None , source = None ):
238238 """
239239
240240 :param var:
@@ -247,17 +247,15 @@ def saveFactory(self, var=None, method=None, qcl=None):
247247 values ['var' ] = ("new_variable" if var else None )
248248 values ['method' ] = ("new_method" if method else None )
249249 values ['qcl' ] = ("new_qcl" if qcl else None )
250+ values ['source' ] = ("new_source" if source else None )
250251 #values['override'] = override
251- return values ['var' ], values ['method' ], values ['qcl' ]#, values['isSave']
252+ return values ['var' ], values ['method' ], values ['qcl' ], values [ 'source' ] #, values['isSave']
252253
253254
254- def save (self , var = None , method = None , qcl = None ):
255+ def save (self ):
255256 """
256257
257- :param var:
258- :param method:
259- :param qcl:
260- :param override:
258+
261259 :return:
262260 """
263261 result = self ._edit_service .save ()
@@ -269,25 +267,26 @@ def save(self, var=None, method=None, qcl=None):
269267 Publisher .sendMessage ("scroll" )
270268 return result
271269
272- def save_as (self , var = None , method = None , qcl = None ):
270+ def save_as (self , var = None , method = None , qcl = None , source = None ):
273271 """
274272
275273 :param var:
276274 :param method:
277275 :param qcl:
276+ :param source:
278277 :param override:
279278 :return:
280279 """
281- result = self ._edit_service .save_as (var = var , method = method , qcl = qcl )
280+ result = self ._edit_service .save_as (var = var , method = method , qcl = qcl , source = source )
282281 if self ._record :
283282 self ._script (
284- "edit_service.save_as(%s, %s, %s)\n " % (self .saveFactory (var , method , qcl )),
283+ "edit_service.save_as(%s, %s, %s, %s )\n " % (self .saveFactory (var , method , qcl , source )),
285284 'black' )
286285 #self._script("edit_service.save(%s, %s, %s, saveAs=%s)\n" % (var, method, qcl, isSave), 'black')
287286 Publisher .sendMessage ("scroll" )
288287 return result
289288
290- def save_appending (self , var = None , method = None , qcl = None , overwrite = False ):
289+ def save_appending (self , var = None , method = None , qcl = None , source = None , overwrite = False ):
291290 """
292291
293292 :param var:
@@ -296,14 +295,14 @@ def save_appending(self, var = None, method =None, qcl = None, overwrite = False
296295 :param override:
297296 :return:
298297 """
299- result = self ._edit_service .save_appending (var = var , method = method , qcl = qcl , overwrite = overwrite )
298+ result = self ._edit_service .save_appending (var = var , method = method , qcl = qcl , source = source , overwrite = overwrite )
300299 if result :
301300 print "Save worked!"
302301
303302 if self ._record :
304303
305304 self ._script (
306- "edit_service.save_appending(%s, %s, %s, " % self .saveFactory (var , method , qcl )+ str (overwrite )+ ")\n " ,
305+ "edit_service.save_appending(%s, %s, %s, %s " % self .saveFactory (var , method , qcl , source )+ str (overwrite )+ ")\n " ,
307306 'black' )
308307 #self._script("edit_service.save(%s, %s, %s, saveAs=%s)\n" % (var, method, qcl, isSave), 'black')
309308 Publisher .sendMessage ("scroll" )
@@ -314,7 +313,7 @@ def save_appending(self, var = None, method =None, qcl = None, overwrite = False
314313 return result
315314
316315
317- def save_existing (self , var = None , method = None , qcl = None ):
316+ def save_existing (self , var = None , method = None , qcl = None , source = None ):
318317 """
319318
320319 :param var:
@@ -323,14 +322,14 @@ def save_existing(self, var=None, method=None, qcl=None):
323322 :param override:
324323 :return:
325324 """
326- result = self ._edit_service .save_existing (var = var , method = method , qcl = qcl )
325+ result = self ._edit_service .save_existing (var = var , method = method , qcl = qcl , source = source )
327326 if result :
328327 print "Save worked!"
329328
330329 if self ._record :
331330
332331 self ._script (
333- "edit_service.save_existing(%s, %s, %s)\n " % (self .saveFactory (var , method , qcl )),
332+ "edit_service.save_existing(%s, %s, %s, %s )\n " % (self .saveFactory (var , method , qcl , source )),
334333 'black' )
335334 #self._script("edit_service.save(%s, %s, %s, saveAs=%s)\n" % (var, method, qcl, isSave), 'black')
336335 Publisher .sendMessage ("scroll" )
@@ -369,6 +368,13 @@ def get_qcl(self, q):
369368 Publisher .sendMessage ("scroll" )
370369
371370 return qcl
371+ def get_source (self , s ):
372+ src = self ._edit_service .get_source (s .id )
373+ if self ._record :
374+ self ._script ('new_src = series_service.get_src_by_id(%s)\n ' % (src .id ))
375+ Publisher .sendMessage ("scroll" )
376+
377+ return src
372378
373379 def get_method (self , m ):
374380 method = self ._edit_service .get_method (m .id )
@@ -434,6 +440,14 @@ def create_variable(self, v):
434440
435441 return var
436442
443+ def create_source (self , src ):
444+ src = self ._edit_service .create_source (src )
445+ if self ._record :
446+ self ._script ('new_variable = series_service.get_soource(%s)\n ' % (src ))
447+ Publisher .sendMessage ("scroll" )
448+
449+ return src
450+
437451 def write_header (self ):
438452 self ._script ("#Uncomment the following commands when running outside ODMTools\n " , 'black' )
439453 self ._script ("#from odmtools.odmservices import EditService, SeriesService\n " , 'black' )
0 commit comments