Package Gnumed :: Package business :: Module gmDemographicRecord
[frames] | no frames]

Source Code for Module Gnumed.business.gmDemographicRecord

   1  # -*- coding: utf8 -*- 
   2  """GNUmed demographics object. 
   3   
   4  This is a patient object intended to let a useful client-side 
   5  API crystallize from actual use in true XP fashion. 
   6   
   7  license: GPL v2 or later 
   8  """ 
   9  #============================================================ 
  10  __author__ = "K.Hilbert <Karsten.Hilbert@gmx.net>, I.Haywood <ihaywood@gnu.org>" 
  11   
  12  # stdlib 
  13  import sys 
  14  import os 
  15  import os.path 
  16  import logging 
  17   
  18   
  19  # GNUmed 
  20  if __name__ == '__main__': 
  21          sys.path.insert(0, '../../') 
  22  from Gnumed.pycommon import gmDispatcher 
  23  from Gnumed.pycommon import gmBusinessDBObject 
  24  from Gnumed.pycommon import gmPG2 
  25  from Gnumed.pycommon import gmTools 
  26   
  27   
  28  _log = logging.getLogger('gm.business') 
  29   
  30  try: 
  31          _ 
  32  except NameError: 
  33          _ = lambda x:x 
  34   
  35  #============================================================ 
  36  # occupation handling 
  37  #------------------------------------------------------------ 
38 -def get_occupations(pk_identity=None):
39 cmd = u""" 40 SELECT * 41 FROM dem.v_person_jobs 42 WHERE pk_identity = %(pk)s 43 ORDER BY l10n_occupation 44 """ 45 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': {'pk': pk_identity}}]) 46 return rows
47 #============================================================ 48 # text+image tags 49 #------------------------------------------------------------ 50 _SQL_get_tag_image = u"SELECT * FROM ref.v_tag_images_no_data WHERE %s" 51
52 -class cTagImage(gmBusinessDBObject.cBusinessDBObject):
53 54 _cmd_fetch_payload = _SQL_get_tag_image % u"pk_tag_image = %s" 55 _cmds_store_payload = [ 56 u""" 57 UPDATE ref.tag_image SET 58 description = gm.nullify_empty_string(%(description)s), 59 filename = gm.nullify_empty_string(%(filename)s) 60 WHERE 61 pk = %(pk_tag_image)s 62 AND 63 xmin = %(xmin_tag_image)s 64 RETURNING 65 pk as pk_tag_image, 66 xmin as xmin_tag_image 67 """ 68 ] 69 _updatable_fields = [u'description', u'filename'] 70 #--------------------------------------------------------
71 - def export_image2file(self, aChunkSize=0, filename=None):
72 73 if self._payload[self._idx['size']] == 0: 74 return None 75 76 if filename is None: 77 suffix = None 78 # preserve original filename extension if available 79 if self._payload[self._idx['filename']] is not None: 80 name, suffix = os.path.splitext(self._payload[self._idx['filename']]) 81 suffix = suffix.strip() 82 if suffix == u'': 83 suffix = None 84 # get unique filename 85 filename = gmTools.get_unique_filename ( 86 prefix = 'gm-tag_image-', 87 suffix = suffix 88 ) 89 90 success = gmPG2.bytea2file ( 91 data_query = { 92 'cmd': u'SELECT substring(image from %(start)s for %(size)s) FROM ref.tag_image WHERE pk = %(pk)s', 93 'args': {'pk': self.pk_obj} 94 }, 95 filename = filename, 96 chunk_size = aChunkSize, 97 data_size = self._payload[self._idx['size']] 98 ) 99 100 if success: 101 return filename 102 103 return None
104 #--------------------------------------------------------
105 - def update_image_from_file(self, filename=None):
106 # sanity check 107 if not (os.access(filename, os.R_OK) and os.path.isfile(filename)): 108 _log.error('[%s] is not a readable file' % filename) 109 return False 110 111 gmPG2.file2bytea ( 112 query = u"UPDATE ref.tag_image SET image = %(data)s::bytea WHERE pk = %(pk)s", 113 filename = filename, 114 args = {'pk': self.pk_obj} 115 ) 116 117 # must update XMIN now ... 118 self.refetch_payload() 119 return True
120 #------------------------------------------------------------
121 -def get_tag_images(order_by=None):
122 if order_by is None: 123 order_by = u'true' 124 else: 125 order_by = 'true ORDER BY %s' % order_by 126 127 cmd = _SQL_get_tag_image % order_by 128 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}], get_col_idx = True) 129 return [ cTagImage(row = {'data': r, 'idx': idx, 'pk_field': 'pk_tag_image'}) for r in rows ]
130 #------------------------------------------------------------
131 -def create_tag_image(description=None, link_obj=None):
132 133 args = {u'desc': description, u'img': u''} 134 cmd = u""" 135 INSERT INTO ref.tag_image ( 136 description, 137 image 138 ) VALUES ( 139 %(desc)s, 140 %(img)s::bytea 141 ) 142 RETURNING pk 143 """ 144 rows, idx = gmPG2.run_rw_queries ( 145 link_obj = link_obj, 146 queries = [{'cmd': cmd, 'args': args}], 147 end_tx = True, 148 return_data = True, 149 get_col_idx = False 150 ) 151 152 return cTagImage(aPK_obj = rows[0]['pk'])
153 #------------------------------------------------------------
154 -def delete_tag_image(tag_image=None):
155 args = {'pk': tag_image} 156 cmd = u""" 157 DELETE FROM ref.tag_image 158 WHERE 159 pk = %(pk)s 160 AND 161 NOT EXISTS ( 162 SELECT 1 163 FROM dem.identity_tag 164 WHERE fk_tag = %(pk)s 165 LIMIT 1 166 ) 167 RETURNING 1 168 """ 169 rows, idx = gmPG2.run_rw_queries(queries = [{'cmd': cmd, 'args': args}], return_data = True) 170 if len(rows) == 0: 171 return False 172 return True
173 174 #============================================================ 175 _SQL_get_identity_tags = u"""SELECT * FROM dem.v_identity_tags WHERE %s""" 176
177 -class cIdentityTag(gmBusinessDBObject.cBusinessDBObject):
178 179 _cmd_fetch_payload = _SQL_get_identity_tags % u"pk_identity_tag = %s" 180 _cmds_store_payload = [ 181 u""" 182 UPDATE dem.identity_tag SET 183 fk_tag = %(pk_tag_image)s, 184 comment = gm.nullify_empty_string(%(comment)s) 185 WHERE 186 pk = %(pk_identity_tag)s 187 AND 188 xmin = %(xmin_identity_tag)s 189 RETURNING 190 pk as pk_identity_tag, 191 xmin as xmin_identity_tag 192 """ 193 ] 194 _updatable_fields = [u'fk_tag', u'comment'] 195 #--------------------------------------------------------
196 - def export_image2file(self, aChunkSize=0, filename=None):
197 198 if self._payload[self._idx['image_size']] == 0: 199 return None 200 201 if filename is None: 202 suffix = None 203 # preserve original filename extension if available 204 if self._payload[self._idx['filename']] is not None: 205 name, suffix = os.path.splitext(self._payload[self._idx['filename']]) 206 suffix = suffix.strip() 207 if suffix == u'': 208 suffix = None 209 # get unique filename 210 filename = gmTools.get_unique_filename ( 211 prefix = 'gm-identity_tag-', 212 suffix = suffix 213 ) 214 215 exported = gmPG2.bytea2file ( 216 data_query = { 217 'cmd': u'SELECT substring(image from %(start)s for %(size)s) FROM ref.tag_image WHERE pk = %(pk)s', 218 'args': {'pk': self._payload[self._idx['pk_tag_image']]} 219 }, 220 filename = filename, 221 chunk_size = aChunkSize, 222 data_size = self._payload[self._idx['image_size']] 223 ) 224 if exported: 225 return filename 226 227 return None
228 #============================================================ 229 #============================================================
230 -def get_countries():
231 cmd = u""" 232 select 233 _(name) as l10n_country, name, code, deprecated 234 from dem.country 235 order by l10n_country""" 236 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}]) 237 return rows
238 #============================================================
239 -def get_country_for_region(region=None):
240 cmd = u""" 241 SELECT code_country, l10n_country FROM dem.v_state WHERE l10n_state = %(region)s 242 union 243 SELECT code_country, l10n_country FROM dem.v_state WHERE state = %(region)s 244 """ 245 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': {'region': region}}]) 246 return rows
247 #============================================================
248 -def delete_province(province=None, delete_urbs=False):
249 250 args = {'prov': province} 251 252 queries = [] 253 if delete_urbs: 254 queries.append ({ 255 'cmd': u""" 256 delete from dem.urb du 257 where 258 du.id_state = %(prov)s 259 and 260 not exists (select 1 from dem.street ds where ds.id_urb = du.id)""", 261 'args': args 262 }) 263 264 queries.append ({ 265 'cmd': u""" 266 delete from dem.state ds 267 where 268 ds.id = %(prov)s 269 and 270 not exists (select 1 from dem.urb du where du.id_state = ds.id)""", 271 'args': args 272 }) 273 274 gmPG2.run_rw_queries(queries = queries) 275 276 return True
277 #------------------------------------------------------------
278 -def create_province(name=None, code=None, country=None):
279 280 args = {'code': code, 'country': country, 'name': name} 281 282 cmd = u"""SELECT EXISTS (SELECT 1 FROM dem.state WHERE name = %(name)s)""" 283 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': args}], get_col_idx = False) 284 285 if rows[0][0]: 286 return 287 288 cmd = u""" 289 INSERT INTO dem.state ( 290 code, country, name 291 ) VALUES ( 292 %(code)s, %(country)s, %(name)s 293 )""" 294 gmPG2.run_rw_queries(queries = [{'cmd': cmd, 'args': args}])
295 #------------------------------------------------------------
296 -def get_provinces():
297 cmd = u""" 298 select 299 l10n_state, l10n_country, state, code_state, code_country, pk_state, country_deprecated 300 from dem.v_state 301 order by l10n_country, l10n_state""" 302 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}]) 303 return rows
304 #============================================================ 305 # address related classes 306 #------------------------------------------------------------
307 -class cAddress(gmBusinessDBObject.cBusinessDBObject):
308 """A class representing an address as an entity in itself. 309 310 We consider addresses to be self-complete "labels" for locations. 311 It does not depend on any people actually living there. Thus 312 an address can get attached to as many people as we want to 313 signify that that is their place of residence/work/... 314 315 This class acts on the address as an entity. Therefore it can 316 modify the address fields. Think carefully about *modifying* 317 addresses attached to people, though. Most times when you think 318 person.modify_address() what you *really* want is as sequence of 319 person.unlink_address(old) and person.link_address(new). 320 321 Modifying an address may or may not be the proper thing to do as 322 it will transparently modify the address for *all* the people to 323 whom it is attached. In many cases you will want to create a *new* 324 address and link it to a person instead of the old address. 325 """ 326 _cmd_fetch_payload = u"select * from dem.v_address where pk_address = %s" 327 _cmds_store_payload = [ 328 u"""UPDATE dem.address SET 329 aux_street = %(notes_street)s, 330 subunit = %(subunit)s, 331 addendum = %(notes_subunit)s, 332 lat_lon = %(lat_lon_street)s 333 WHERE 334 id = %(pk_address)s 335 AND 336 xmin = %(xmin_address)s 337 RETURNING 338 xmin AS xmin_address""" 339 ] 340 _updatable_fields = [ 341 'notes_street', 342 'subunit', 343 'notes_subunit', 344 'lat_lon_address' 345 ] 346 #--------------------------------------------------------
347 - def format(self, single_line=False, verbose=False, show_type=False):
348 if single_line: 349 return format_address_single_line(address = self, show_type = False, verbose = verbose) 350 return format_address(address = self, show_type = False)
351 #------------------------------------------------------------
352 -def address_exists(country=None, state=None, urb=None, postcode=None, street=None, number=None, subunit=None):
353 354 cmd = u"""SELECT dem.address_exists(%(country)s, %(state)s, %(urb)s, %(postcode)s, %(street)s, %(number)s, %(subunit)s)""" 355 args = { 356 'country': country, 357 'state': state, 358 'urb': urb, 359 'postcode': postcode, 360 'street': street, 361 'number': number, 362 'subunit': subunit 363 } 364 365 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': args}]) 366 if rows[0][0] is None: 367 _log.debug('address does not exist') 368 for key, val in args.items(): 369 _log.debug('%s: %s', key, val) 370 return None 371 372 return rows[0][0]
373 #------------------------------------------------------------
374 -def create_address(country=None, state=None, urb=None, suburb=None, postcode=None, street=None, number=None, subunit=None):
375 376 if suburb is not None: 377 suburb = gmTools.none_if(suburb.strip(), u'') 378 379 pk_address = address_exists ( 380 country = country, 381 state = state, 382 urb = urb, 383 # suburb = suburb, 384 postcode = postcode, 385 street = street, 386 number = number, 387 subunit = subunit 388 ) 389 if pk_address is not None: 390 return cAddress(aPK_obj = pk_address) 391 392 cmd = u""" 393 SELECT dem.create_address ( 394 %(number)s, 395 %(street)s, 396 %(postcode)s, 397 %(urb)s, 398 %(state)s, 399 %(country)s, 400 %(subunit)s 401 )""" 402 args = { 403 'number': number, 404 'street': street, 405 'postcode': postcode, 406 'urb': urb, 407 'state': state, 408 'country': country, 409 'subunit': subunit 410 } 411 queries = [{'cmd': cmd, 'args': args}] 412 413 rows, idx = gmPG2.run_rw_queries(queries = queries, return_data = True) 414 adr = cAddress(aPK_obj = rows[0][0]) 415 416 if suburb is not None: 417 queries = [{ 418 # CAVE: suburb will be ignored if there already is one 419 'cmd': u"UPDATE dem.street SET suburb = %(suburb)s WHERE id = %(pk_street)s AND suburb IS NULL", 420 'args': {'suburb': suburb, 'pk_street': adr['pk_street']} 421 }] 422 rows, idx = gmPG2.run_rw_queries(queries = queries) 423 424 return adr
425 #------------------------------------------------------------
426 -def delete_address(pk_address=None):
427 cmd = u""" 428 DELETE FROM dem.address 429 WHERE 430 id = %(pk)s 431 AND 432 NOT EXISTS (( 433 SELECT 1 FROM dem.org_unit WHERE fk_address = %(pk)s LIMIT 1 434 ) UNION ( 435 SELECT 1 FROM dem.lnk_identity2comm WHERE fk_address = %(pk)s LIMIT 1 436 ) UNION ( 437 SELECT 1 FROM dem.lnk_person_org_address WHERE id_address = %(pk)s LIMIT 1 438 )) 439 """ 440 rows, idx = gmPG2.run_rw_queries(queries = [{'cmd': cmd, 'args': {'pk': pk_address}}]) 441 return True
442 #------------------------------------------------------------
443 -def format_address_single_line(address=None, verbose=False, show_type=False):
444 data = { 445 'pk_adr': address['pk_address'], 446 'street': address['street'], 447 'notes_street': gmTools.coalesce(address['notes_street'], u'', u' (%s)'), 448 'number': address['number'], 449 'subunit': gmTools.coalesce(address['subunit'], u'', u'/%s'), 450 'notes_subunit': gmTools.coalesce(address['notes_subunit'], u'', u' (%s)'), 451 'zip': address['postcode'], 452 'urb': address['urb'], 453 'suburb': gmTools.coalesce(address['suburb'], u'', u' (%s)'), 454 'l10n_state': address['l10n_state'], 455 'code_state': address['code_state'], 456 'l10n_country': address['l10n_country'], 457 'code_country': address['code_country'] 458 } 459 if show_type: 460 data['type'] = address['l10n_address_type'] 461 462 if verbose: 463 if show_type: 464 template = _('%(type)s: %(street)s %(number)s%(subunit)s, %(zip)s %(urb)s %(suburb)s, %(code_state)s, %(code_country)s') 465 else: 466 template = _('%(street)s %(number)s%(subunit)s, %(zip)s %(urb)s %(suburb)s, %(code_state)s, %(code_country)s') 467 else: 468 if show_type: 469 template = _('%(type)s: %(street)s %(number)s%(subunit)s, %(zip)s %(urb)s, %(code_state)s, %(code_country)s') 470 else: 471 template = _('%(street)s %(number)s%(subunit)s, %(zip)s %(urb)s, %(code_state)s, %(code_country)s') 472 473 return template % data
474 #------------------------------------------------------------
475 -def format_address(address=None, show_type=False):
476 data = { 477 'pk_adr': address['pk_address'], 478 'street': address['street'], 479 'notes_street': gmTools.coalesce(address['notes_street'], u'', u' (%s)'), 480 'number': address['number'], 481 'subunit': gmTools.coalesce(address['subunit'], u'', u'/%s'), 482 'notes_subunit': gmTools.coalesce(address['notes_subunit'], u'', u' (%s)'), 483 'zip': address['postcode'], 484 'urb': address['urb'], 485 'suburb': gmTools.coalesce(address['suburb'], u'', u' (%s)'), 486 'l10n_state': address['l10n_state'], 487 'code_state': address['code_state'], 488 'l10n_country': address['l10n_country'], 489 'code_country': address['code_country'] 490 } 491 if show_type: 492 data['type'] = address['l10n_address_type'] 493 template = _( 494 'Address (%(type)s) [#%(pk_adr)s]\n' 495 ' Street: %(street)s%(notes_street)s\n' 496 ' Number/Unit: %(number)s%(subunit)s%(notes_subunit)s\n' 497 ' Location: %(zip)s %(urb)s%(suburb)s\n' 498 ' Region: %(l10n_state)s, %(code_state)s\n' 499 ' Country: %(l10n_country)s, %(code_country)s' 500 ) 501 else: 502 template = _( 503 'Address [#%(pk_adr)s]\n' 504 ' Street: %(street)s%(notes_street)s\n' 505 ' Number/Unit: %(number)s%(subunit)s%(notes_subunit)s\n' 506 ' Location: %(zip)s %(urb)s%(suburb)s\n' 507 ' Region: %(l10n_state)s, %(code_state)s\n' 508 ' Country: %(l10n_country)s, %(code_country)s' 509 ) 510 txt = template % data 511 return txt.split('\n')
512 #------------------------------------------------------------
513 -def get_address_types(identity=None):
514 cmd = u'select id as pk, name, _(name) as l10n_name from dem.address_type' 515 rows, idx = gmPG2.run_rw_queries(queries=[{'cmd': cmd}]) 516 return rows
517 #------------------------------------------------------------
518 -def get_addresses(order_by=None):
519 520 if order_by is None: 521 order_by = u'' 522 else: 523 order_by = u'ORDER BY %s' % order_by 524 525 cmd = u"SELECT * FROM dem.v_address %s" % order_by 526 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}], get_col_idx = True) 527 return [ cAddress(row = {'data': r, 'idx': idx, 'pk_field': u'pk_address'}) for r in rows ]
528 #------------------------------------------------------------
529 -def get_address_from_patient_address_pk(pk_patient_address=None):
530 cmd = u""" 531 SELECT * FROM dem.v_address WHERE 532 pk_address = ( 533 SELECT id_address 534 FROM dem.lnk_person_org_address 535 WHERE id = %(pk_pat_adr)s 536 ) 537 """ 538 args = {'pk_pat_adr': pk_patient_address} 539 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': args}], get_col_idx = True) 540 if len(rows) == 0: 541 return None 542 return cAddress(row = {'data': rows[0], 'idx': idx, 'pk_field': u'pk_address'})
543 544 #===================================================================
545 -def get_patient_address(pk_patient_address=None):
546 cmd = u'SELECT * FROM dem.v_pat_addresses WHERE pk_lnk_person_org_address = %(pk)s' 547 args = {'pk': pk_patient_address} 548 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': args}], get_col_idx = True) 549 if len(rows) == 0: 550 return None 551 return cPatientAddress(row = {'data': rows[0], 'idx': idx, 'pk_field': u'pk_address'})
552 #-------------------------------------------------------------------
553 -def get_patient_address_by_type(pk_patient=None, adr_type=None):
554 cmd = u'SELECT * FROM dem.v_pat_addresses WHERE pk_identity = %(pat)s AND address_type = %(typ)s' 555 args = {'pat': pk_patient, 'typ': adr_type} 556 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd, 'args': args}], get_col_idx = True) 557 if len(rows) == 0: 558 return None 559 return cPatientAddress(row = {'data': rows[0], 'idx': idx, 'pk_field': u'pk_address'})
560 #-------------------------------------------------------------------
561 -class cPatientAddress(gmBusinessDBObject.cBusinessDBObject):
562 563 _cmd_fetch_payload = u"SELECT * FROM dem.v_pat_addresses WHERE pk_address = %s" 564 _cmds_store_payload = [ 565 u"""UPDATE dem.lnk_person_org_address SET 566 id_type = %(pk_address_type)s 567 WHERE 568 id = %(pk_lnk_person_org_address)s 569 AND 570 xmin = %(xmin_lnk_person_org_address)s 571 RETURNING 572 xmin AS xmin_lnk_person_org_address 573 """ 574 ] 575 _updatable_fields = ['pk_address_type'] 576 #---------------------------------------------------------------
577 - def get_identities(self, same_lastname=False):
578 pass
579 #--------------------------------------------------------
580 - def format(self, single_line=False, verbose=False, show_type=True):
581 if single_line: 582 return format_address_single_line(address = self, verbose = verbose, show_type = show_type) 583 txt = format_address(address = self, show_type = show_type) 584 return txt
585 #=================================================================== 586 # communication channels API 587 #-------------------------------------------------------------------
588 -class cCommChannel(gmBusinessDBObject.cBusinessDBObject):
589 590 _cmd_fetch_payload = u"SELECT * FROM dem.v_person_comms WHERE pk_lnk_identity2comm = %s" 591 _cmds_store_payload = [ 592 u"""UPDATE dem.lnk_identity2comm SET 593 --fk_address = %(pk_address)s, 594 fk_type = dem.create_comm_type(%(comm_type)s), 595 url = %(url)s, 596 is_confidential = %(is_confidential)s, 597 comment = gm.nullify_empty_string(%(comment)s) 598 WHERE 599 pk = %(pk_lnk_identity2comm)s 600 AND 601 xmin = %(xmin_lnk_identity2comm)s 602 RETURNING 603 xmin AS xmin_lnk_identity2comm 604 """ 605 ] 606 _updatable_fields = [ 607 #'pk_address', 608 'url', 609 'comm_type', 610 'is_confidential', 611 'comment' 612 ]
613 #-------------------------------------------------------------------
614 -class cOrgCommChannel(gmBusinessDBObject.cBusinessDBObject):
615 616 _cmd_fetch_payload = u"SELECT * FROM dem.v_org_unit_comms WHERE pk_lnk_org_unit2comm = %s" 617 _cmds_store_payload = [ 618 u"""UPDATE dem.lnk_org_unit2comm SET 619 fk_type = dem.create_comm_type(%(comm_type)s), 620 url = %(url)s, 621 is_confidential = %(is_confidential)s 622 WHERE 623 pk = %(pk_lnk_org_unit2comm)s 624 AND 625 xmin = %(xmin_lnk_org_unit2comm)s 626 RETURNING 627 xmin AS xmin_lnk_org_unit2comm 628 """ 629 ] 630 _updatable_fields = [ 631 'url', 632 'comm_type', 633 'is_confidential' 634 ]
635 #-------------------------------------------------------------------
636 -def create_comm_channel(comm_medium=None, url=None, is_confidential=False, pk_channel_type=None, pk_identity=None, pk_org_unit=None):
637 """Create a communications channel for a patient.""" 638 639 if url is None: 640 return None 641 642 args = { 643 'url': url, 644 'secret': is_confidential, 645 'pk_type': pk_channel_type, 646 'type': comm_medium 647 } 648 649 if pk_identity is not None: 650 args['pk_owner'] = pk_identity 651 tbl = u'dem.lnk_identity2comm' 652 col = u'fk_identity' 653 view = u'dem.v_person_comms' 654 view_pk = u'pk_lnk_identity2comm' 655 channel_class = cCommChannel 656 if pk_org_unit is not None: 657 args['pk_owner'] = pk_org_unit 658 tbl = u'dem.lnk_org_unit2comm' 659 col = u'fk_org_unit' 660 view = u'dem.v_org_unit_comms' 661 view_pk = u'pk_lnk_org_unit2comm' 662 channel_class = cOrgCommChannel 663 664 if pk_channel_type is None: 665 cmd = u"""INSERT INTO %s ( 666 %s, 667 url, 668 fk_type, 669 is_confidential 670 ) VALUES ( 671 %%(pk_owner)s, 672 %%(url)s, 673 dem.create_comm_type(%%(type)s), 674 %%(secret)s 675 )""" % (tbl, col) 676 else: 677 cmd = u"""INSERT INTO %s ( 678 %s, 679 url, 680 fk_type, 681 is_confidential 682 ) VALUES ( 683 %%(pk_owner)s, 684 %%(url)s, 685 %%(pk_type)s, 686 %%(secret)s 687 )""" % (tbl, col) 688 689 queries = [{'cmd': cmd, 'args': args}] 690 cmd = u"SELECT * FROM %s WHERE %s = currval(pg_get_serial_sequence('%s', 'pk'))" % (view, view_pk, tbl) 691 queries.append({'cmd': cmd}) 692 693 rows, idx = gmPG2.run_rw_queries(queries = queries, return_data = True, get_col_idx = True) 694 695 if pk_identity is not None: 696 return cCommChannel(row = {'pk_field': view_pk, 'data': rows[0], 'idx': idx}) 697 698 return channel_class(row = {'pk_field': view_pk, 'data': rows[0], 'idx': idx})
699 #-------------------------------------------------------------------
700 -def delete_comm_channel(pk=None, pk_patient=None, pk_org_unit=None):
701 if pk_patient is not None: 702 query = { 703 'cmd': u"DELETE FROM dem.lnk_identity2comm WHERE pk = %(pk)s AND fk_identity = %(pat)s", 704 'args': {'pk': pk, 'pat': pk_patient} 705 } 706 if pk_org_unit is not None: 707 query = { 708 'cmd': u"DELETE FROM dem.lnk_org_unit2comm WHERE pk = %(pk)s AND fk_org_unit = %(unit)s", 709 'args': {'pk': pk, 'unit': pk_org_unit} 710 } 711 gmPG2.run_rw_queries(queries = [query])
712 #-------------------------------------------------------------------
713 -def get_comm_channel_types():
714 cmd = u"SELECT pk, _(description) AS l10n_description, description FROM dem.enum_comm_types" 715 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}], get_col_idx = False) 716 return rows
717 #-------------------------------------------------------------------
718 -def delete_comm_channel_type(pk_channel_type=None):
719 cmd = u""" 720 DELETE FROM dem.enum_comm_types 721 WHERE 722 pk = %(pk)s 723 AND NOT EXISTS ( 724 SELECT 1 FROM dem.lnk_identity2comm WHERE fk_type = %(pk)s 725 ) 726 AND NOT EXISTS ( 727 SELECT 1 FROM dem.lnk_org_unit2comm WHERE fk_type = %(pk)s 728 ) 729 """ 730 gmPG2.run_rw_queries(queries = [{'cmd': cmd, 'args': {'pk': pk_channel_type}}]) 731 return True
732 #=================================================================== 733 #------------------------------------------------------------------- 734
735 -class cOrg (gmBusinessDBObject.cBusinessDBObject):
736 """ 737 Organisations 738 739 This is also the common ancestor of cIdentity, self._table is used to 740 hide the difference. 741 The aim is to be able to sanely write code which doesn't care whether 742 its talking to an organisation or an individual""" 743 _table = "org" 744 745 _cmd_fetch_payload = "select *, xmin from dem.org where id=%s" 746 _cmds_lock_rows_for_update = ["select 1 from dem.org where id=%(id)s and xmin=%(xmin)s"] 747 _cmds_store_payload = [ 748 """update dem.org set 749 description=%(description)s, 750 id_category=(select id from dem.org_category where description=%(occupation)s) 751 where id=%(id)s""", 752 "select xmin from dem.org where id=%(id)s" 753 ] 754 _updatable_fields = ["description", "occupation"] 755 _service = 'personalia' 756 #------------------------------------------------------------------
757 - def cleanup (self):
758 pass
759 #------------------------------------------------------------------
760 - def export_demographics (self):
761 if not self.__cache.has_key ('addresses'): 762 self['addresses'] 763 if not self.__cache.has_key ('comms'): 764 self['comms'] 765 return self.__cache
766 #--------------------------------------------------------------------
767 - def get_members (self):
768 """ 769 Returns a list of (address dict, cIdentity) tuples 770 """ 771 cmd = """select 772 vba.id, 773 vba.number, 774 vba.addendum, 775 vba.street, 776 vba.urb, 777 vba.postcode, 778 at.name, 779 lpoa.id_type, 780 vbp.pk_identity, 781 title, 782 firstnames, 783 lastnames, 784 dob, 785 cob, 786 gender, 787 pupic, 788 pk_marital_status, 789 marital_status, 790 karyotype, 791 xmin_identity, 792 preferred 793 from 794 dem.v_basic_address vba, 795 dem.lnk_person_org_address lpoa, 796 dem.address_type at, 797 dem.v_basic_person vbp 798 where 799 lpoa.id_address = vba.id 800 and lpoa.id_type = at.id 801 and lpoa.id_identity = vbp.pk_identity 802 and lpoa.id_org = %%s 803 """ 804 805 rows, idx = gmPG.run_ro_query('personalia', cmd, 1, self.getId ()) 806 if rows is None: 807 return [] 808 elif len(rows) == 0: 809 return [] 810 else: 811 return [({'pk':i[0], 'number':i[1], 'addendum':i[2], 'street':i[3], 'city':i[4], 'postcode':i[5], 'type':i[6], 'id_type':i[7]}, cIdentity (row = {'data':i[8:], 'id':idx[8:], 'pk_field':'id'})) for i in rows]
812 #------------------------------------------------------------
813 - def set_member (self, person, address):
814 """ 815 Binds a person to this organisation at this address. 816 person is a cIdentity object 817 address is a dict of {'number', 'street', 'addendum', 'city', 'postcode', 'type'} 818 type is one of the IDs returned by getAddressTypes 819 """ 820 cmd = "insert into dem.lnk_person_org_address (id_type, id_address, id_org, id_identity) values (%(type)s, dem.create_address (%(number)s, %(addendum)s, %(street)s, %(city)s, %(postcode)s), %(org_id)s, %(pk_identity)s)" 821 address['pk_identity'] = person['pk_identity'] 822 address['org_id'] = self.getId() 823 if not id_addr: 824 return (False, None) 825 return gmPG.run_commit2 ('personalia', [(cmd, [address])])
826 #------------------------------------------------------------ 830 #----------------------------------------------------------------------
831 - def getId (self):
832 """ 833 Hide the difference between org.id and v_basic_person.pk_identity 834 """ 835 return self['id']
836 #==============================================================================
837 -def get_time_tuple (mx):
838 """ 839 wrap mx.DateTime brokenness 840 Returns 9-tuple for use with pyhon time functions 841 """ 842 return [ int(x) for x in str(mx).split(' ')[0].split('-') ] + [0,0,0, 0,0,0]
843 #----------------------------------------------------------------
844 -def getAddressTypes():
845 """Gets a dict matching address types to their ID""" 846 row_list = gmPG.run_ro_query('personalia', "select name, id from dem.address_type") 847 if row_list is None: 848 return {} 849 if len(row_list) == 0: 850 return {} 851 return dict (row_list)
852 #----------------------------------------------------------------
853 -def getMaritalStatusTypes():
854 """Gets a dictionary matching marital status types to their internal ID""" 855 row_list = gmPG.run_ro_query('personalia', "select name, pk from dem.marital_status") 856 if row_list is None: 857 return {} 858 if len(row_list) == 0: 859 return {} 860 return dict(row_list)
861 #------------------------------------------------------------------
862 -def getRelationshipTypes():
863 """Gets a dictionary of relationship types to internal id""" 864 row_list = gmPG.run_ro_query('personalia', "select description, id from dem.relation_types") 865 if row_list is None: 866 return None 867 if len (row_list) == 0: 868 return None 869 return dict(row_list)
870 871 #----------------------------------------------------------------
872 -def getUrb (id_urb):
873 cmd = """ 874 select 875 dem.state.name, 876 dem.urb.postcode 877 from 878 dem.urb, 879 dem.state 880 where 881 dem.urb.id = %s and 882 dem.urb.id_state = dem.state.id""" 883 row_list = gmPG.run_ro_query('personalia', cmd, None, id_urb) 884 if not row_list: 885 return None 886 else: 887 return (row_list[0][0], row_list[0][1])
888
889 -def getStreet (id_street):
890 cmd = """ 891 select 892 dem.state.name, 893 coalesce (dem.street.postcode, dem.urb.postcode), 894 dem.urb.name 895 from 896 dem.urb, 897 dem.state, 898 dem.street 899 where 900 dem.street.id = %s and 901 dem.street.id_urb = dem.urb.id and 902 dem.urb.id_state = dem.state.id 903 """ 904 row_list = gmPG.run_ro_query('personalia', cmd, None, id_street) 905 if not row_list: 906 return None 907 else: 908 return (row_list[0][0], row_list[0][1], row_list[0][2])
909
910 -def getCountry (country_code):
911 row_list = gmPG.run_ro_query('personalia', "select name from dem.country where code = %s", None, country_code) 912 if not row_list: 913 return None 914 else: 915 return row_list[0][0]
916 #-------------------------------------------------------------------------------
917 -def get_town_data (town):
918 row_list = gmPG.run_ro_query ('personalia', """ 919 select 920 dem.urb.postcode, 921 dem.state.code, 922 dem.state.name, 923 dem.country.code, 924 dem.country.name 925 from 926 dem.urb, 927 dem.state, 928 dem.country 929 where 930 dem.urb.name = %s and 931 dem.urb.id_state = dem.state.id and 932 dem.state.country = dem.country.code""", None, town) 933 if not row_list: 934 return (None, None, None, None, None) 935 else: 936 return tuple (row_list[0])
937 #============================================================ 938 # callbacks 939 #------------------------------------------------------------
940 -def _post_patient_selection(**kwargs):
941 print "received post_patient_selection notification" 942 print kwargs['kwds']
943 #============================================================ 944 945 #============================================================ 946 # main 947 #------------------------------------------------------------ 948 if __name__ == "__main__": 949 950 if len(sys.argv) < 2: 951 sys.exit() 952 953 if sys.argv[1] != 'test': 954 sys.exit() 955 956 import random 957 #--------------------------------------------------------
958 - def test_address_exists():
959 960 addresses = [ 961 { 962 'country': 'Germany', 963 'state': 'Sachsen', 964 'urb': 'Leipzig', 965 'postcode': '04318', 966 'street': u'Cunnersdorfer Strasse', 967 'number': '11' 968 }, 969 { 970 'country': 'DE', 971 'state': 'SN', 972 'urb': 'Leipzig', 973 'postcode': '04317', 974 'street': u'Riebeckstraße', 975 'number': '65', 976 'subunit': 'Parterre' 977 }, 978 { 979 'country': 'DE', 980 'state': 'SN', 981 'urb': 'Leipzig', 982 'postcode': '04317', 983 'street': u'Riebeckstraße', 984 'number': '65', 985 'subunit': '1. Stock' 986 }, 987 { 988 'country': 'DE', 989 'state': 'SN', 990 'urb': 'Leipzig', 991 'postcode': '04317', 992 'street': u'Riebeckstraße', 993 'number': '65', 994 'subunit': '1. Stock' 995 }, 996 { 997 # 'country': 'DE', 998 # 'state': 'SN', 999 'urb': 'Leipzig', 1000 'postcode': '04317', 1001 'street': u'Riebeckstraße', 1002 'number': '65', 1003 'subunit': '1. Stock' 1004 }, 1005 ] 1006 1007 for adr in addresses: 1008 print adr 1009 exists = address_exists(**adr) 1010 if exists is None: 1011 print "address does not exist" 1012 else: 1013 print "address exists, primary key:", exists
1014 1015 #--------------------------------------------------------
1016 - def test_create_address():
1017 address = create_address ( 1018 country ='DE', 1019 state ='SN', 1020 urb ='Leipzig', 1021 suburb ='Sellerhausen', 1022 postcode ='04318', 1023 street = u'Cunnersdorfer Strasse', 1024 number = '11' 1025 # ,notes_subunit = '4.Stock rechts' 1026 ) 1027 print "created existing address" 1028 print address.format() 1029 1030 su = str(random.random()) 1031 1032 address = create_address ( 1033 country ='DE', 1034 state = 'SN', 1035 urb ='Leipzig', 1036 suburb ='Sellerhausen', 1037 postcode ='04318', 1038 street = u'Cunnersdorfer Strasse', 1039 number = '11', 1040 # notes_subunit = '4.Stock rechts', 1041 subunit = su 1042 ) 1043 print "created new address with subunit", su 1044 print address 1045 print address.format() 1046 print "deleted address:", delete_address(pk_address = address['pk_address'])
1047 #--------------------------------------------------------
1048 - def test_get_countries():
1049 for c in get_countries(): 1050 print c
1051 #--------------------------------------------------------
1052 - def test_get_country_for_region():
1053 region = raw_input("Please enter a region: ") 1054 print "country for region [%s] is: %s" % (region, get_country_for_region(region = region))
1055 #--------------------------------------------------------
1056 - def test_delete_tag():
1057 if delete_tag_image(tag_image = 9999): 1058 print "deleted tag 9999" 1059 else: 1060 print "did not delete tag 9999" 1061 if delete_tag_image(tag_image = 1): 1062 print "deleted tag 1" 1063 else: 1064 print "did not delete tag 1"
1065 #--------------------------------------------------------
1066 - def test_tag_images():
1067 tag = cTagImage(aPK_obj = 1) 1068 print tag
1069 #print get_tag_images() 1070 #--------------------------------------------------------
1071 - def test_get_billing_address():
1072 print get_patient_address_by_type(pk_patient = 12, adr_type = u'billing')
1073 #-------------------------------------------------------- 1074 #gmPG2.get_connection() 1075 1076 #test_address_exists() 1077 #test_create_address() 1078 #test_get_countries() 1079 #test_get_country_for_region() 1080 #test_delete_tag() 1081 test_tag_images() 1082 #test_get_billing_address() 1083 1084 sys.exit() 1085 #============================================================ 1086