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 - def _get_address(self):
587 return cAddress(aPK_obj = self._payload[self._idx['pk_address']])
588 589 address = property(_get_address, lambda x:x)
590 #=================================================================== 591 # communication channels API 592 #-------------------------------------------------------------------
593 -class cCommChannel(gmBusinessDBObject.cBusinessDBObject):
594 595 _cmd_fetch_payload = u"SELECT * FROM dem.v_person_comms WHERE pk_lnk_identity2comm = %s" 596 _cmds_store_payload = [ 597 u"""UPDATE dem.lnk_identity2comm SET 598 --fk_address = %(pk_address)s, 599 fk_type = dem.create_comm_type(%(comm_type)s), 600 url = %(url)s, 601 is_confidential = %(is_confidential)s, 602 comment = gm.nullify_empty_string(%(comment)s) 603 WHERE 604 pk = %(pk_lnk_identity2comm)s 605 AND 606 xmin = %(xmin_lnk_identity2comm)s 607 RETURNING 608 xmin AS xmin_lnk_identity2comm 609 """ 610 ] 611 _updatable_fields = [ 612 #'pk_address', 613 'url', 614 'comm_type', 615 'is_confidential', 616 'comment' 617 ]
618 #-------------------------------------------------------------------
619 -class cOrgCommChannel(gmBusinessDBObject.cBusinessDBObject):
620 621 _cmd_fetch_payload = u"SELECT * FROM dem.v_org_unit_comms WHERE pk_lnk_org_unit2comm = %s" 622 _cmds_store_payload = [ 623 u"""UPDATE dem.lnk_org_unit2comm SET 624 fk_type = dem.create_comm_type(%(comm_type)s), 625 url = %(url)s, 626 is_confidential = %(is_confidential)s 627 WHERE 628 pk = %(pk_lnk_org_unit2comm)s 629 AND 630 xmin = %(xmin_lnk_org_unit2comm)s 631 RETURNING 632 xmin AS xmin_lnk_org_unit2comm 633 """ 634 ] 635 _updatable_fields = [ 636 'url', 637 'comm_type', 638 'is_confidential' 639 ]
640 #-------------------------------------------------------------------
641 -def create_comm_channel(comm_medium=None, url=None, is_confidential=False, pk_channel_type=None, pk_identity=None, pk_org_unit=None):
642 """Create a communications channel for a patient.""" 643 644 if url is None: 645 return None 646 647 args = { 648 'url': url, 649 'secret': is_confidential, 650 'pk_type': pk_channel_type, 651 'type': comm_medium 652 } 653 654 if pk_identity is not None: 655 args['pk_owner'] = pk_identity 656 tbl = u'dem.lnk_identity2comm' 657 col = u'fk_identity' 658 view = u'dem.v_person_comms' 659 view_pk = u'pk_lnk_identity2comm' 660 channel_class = cCommChannel 661 if pk_org_unit is not None: 662 args['pk_owner'] = pk_org_unit 663 tbl = u'dem.lnk_org_unit2comm' 664 col = u'fk_org_unit' 665 view = u'dem.v_org_unit_comms' 666 view_pk = u'pk_lnk_org_unit2comm' 667 channel_class = cOrgCommChannel 668 669 if pk_channel_type is None: 670 cmd = u"""INSERT INTO %s ( 671 %s, 672 url, 673 fk_type, 674 is_confidential 675 ) VALUES ( 676 %%(pk_owner)s, 677 %%(url)s, 678 dem.create_comm_type(%%(type)s), 679 %%(secret)s 680 )""" % (tbl, col) 681 else: 682 cmd = u"""INSERT INTO %s ( 683 %s, 684 url, 685 fk_type, 686 is_confidential 687 ) VALUES ( 688 %%(pk_owner)s, 689 %%(url)s, 690 %%(pk_type)s, 691 %%(secret)s 692 )""" % (tbl, col) 693 694 queries = [{'cmd': cmd, 'args': args}] 695 cmd = u"SELECT * FROM %s WHERE %s = currval(pg_get_serial_sequence('%s', 'pk'))" % (view, view_pk, tbl) 696 queries.append({'cmd': cmd}) 697 698 rows, idx = gmPG2.run_rw_queries(queries = queries, return_data = True, get_col_idx = True) 699 700 if pk_identity is not None: 701 return cCommChannel(row = {'pk_field': view_pk, 'data': rows[0], 'idx': idx}) 702 703 return channel_class(row = {'pk_field': view_pk, 'data': rows[0], 'idx': idx})
704 #-------------------------------------------------------------------
705 -def delete_comm_channel(pk=None, pk_patient=None, pk_org_unit=None):
706 if pk_patient is not None: 707 query = { 708 'cmd': u"DELETE FROM dem.lnk_identity2comm WHERE pk = %(pk)s AND fk_identity = %(pat)s", 709 'args': {'pk': pk, 'pat': pk_patient} 710 } 711 if pk_org_unit is not None: 712 query = { 713 'cmd': u"DELETE FROM dem.lnk_org_unit2comm WHERE pk = %(pk)s AND fk_org_unit = %(unit)s", 714 'args': {'pk': pk, 'unit': pk_org_unit} 715 } 716 gmPG2.run_rw_queries(queries = [query])
717 #-------------------------------------------------------------------
718 -def get_comm_channel_types():
719 cmd = u"SELECT pk, _(description) AS l10n_description, description FROM dem.enum_comm_types" 720 rows, idx = gmPG2.run_ro_queries(queries = [{'cmd': cmd}], get_col_idx = False) 721 return rows
722 #-------------------------------------------------------------------
723 -def delete_comm_channel_type(pk_channel_type=None):
724 cmd = u""" 725 DELETE FROM dem.enum_comm_types 726 WHERE 727 pk = %(pk)s 728 AND NOT EXISTS ( 729 SELECT 1 FROM dem.lnk_identity2comm WHERE fk_type = %(pk)s 730 ) 731 AND NOT EXISTS ( 732 SELECT 1 FROM dem.lnk_org_unit2comm WHERE fk_type = %(pk)s 733 ) 734 """ 735 gmPG2.run_rw_queries(queries = [{'cmd': cmd, 'args': {'pk': pk_channel_type}}]) 736 return True
737 #=================================================================== 738 #------------------------------------------------------------------- 739
740 -class cOrg (gmBusinessDBObject.cBusinessDBObject):
741 """ 742 Organisations 743 744 This is also the common ancestor of cIdentity, self._table is used to 745 hide the difference. 746 The aim is to be able to sanely write code which doesn't care whether 747 its talking to an organisation or an individual""" 748 _table = "org" 749 750 _cmd_fetch_payload = "select *, xmin from dem.org where id=%s" 751 _cmds_lock_rows_for_update = ["select 1 from dem.org where id=%(id)s and xmin=%(xmin)s"] 752 _cmds_store_payload = [ 753 """update dem.org set 754 description=%(description)s, 755 id_category=(select id from dem.org_category where description=%(occupation)s) 756 where id=%(id)s""", 757 "select xmin from dem.org where id=%(id)s" 758 ] 759 _updatable_fields = ["description", "occupation"] 760 _service = 'personalia' 761 #------------------------------------------------------------------
762 - def cleanup (self):
763 pass
764 #------------------------------------------------------------------
765 - def export_demographics (self):
766 if not self.__cache.has_key ('addresses'): 767 self['addresses'] 768 if not self.__cache.has_key ('comms'): 769 self['comms'] 770 return self.__cache
771 #--------------------------------------------------------------------
772 - def get_members (self):
773 """ 774 Returns a list of (address dict, cIdentity) tuples 775 """ 776 cmd = """select 777 vba.id, 778 vba.number, 779 vba.addendum, 780 vba.street, 781 vba.urb, 782 vba.postcode, 783 at.name, 784 lpoa.id_type, 785 vbp.pk_identity, 786 title, 787 firstnames, 788 lastnames, 789 dob, 790 cob, 791 gender, 792 pupic, 793 pk_marital_status, 794 marital_status, 795 karyotype, 796 xmin_identity, 797 preferred 798 from 799 dem.v_basic_address vba, 800 dem.lnk_person_org_address lpoa, 801 dem.address_type at, 802 dem.v_basic_person vbp 803 where 804 lpoa.id_address = vba.id 805 and lpoa.id_type = at.id 806 and lpoa.id_identity = vbp.pk_identity 807 and lpoa.id_org = %%s 808 """ 809 810 rows, idx = gmPG.run_ro_query('personalia', cmd, 1, self.getId ()) 811 if rows is None: 812 return [] 813 elif len(rows) == 0: 814 return [] 815 else: 816 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]
817 #------------------------------------------------------------
818 - def set_member (self, person, address):
819 """ 820 Binds a person to this organisation at this address. 821 person is a cIdentity object 822 address is a dict of {'number', 'street', 'addendum', 'city', 'postcode', 'type'} 823 type is one of the IDs returned by getAddressTypes 824 """ 825 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)" 826 address['pk_identity'] = person['pk_identity'] 827 address['org_id'] = self.getId() 828 if not id_addr: 829 return (False, None) 830 return gmPG.run_commit2 ('personalia', [(cmd, [address])])
831 #------------------------------------------------------------ 835 #----------------------------------------------------------------------
836 - def getId (self):
837 """ 838 Hide the difference between org.id and v_basic_person.pk_identity 839 """ 840 return self['id']
841 #==============================================================================
842 -def get_time_tuple (mx):
843 """ 844 wrap mx.DateTime brokenness 845 Returns 9-tuple for use with pyhon time functions 846 """ 847 return [ int(x) for x in str(mx).split(' ')[0].split('-') ] + [0,0,0, 0,0,0]
848 #----------------------------------------------------------------
849 -def getAddressTypes():
850 """Gets a dict matching address types to their ID""" 851 row_list = gmPG.run_ro_query('personalia', "select name, id from dem.address_type") 852 if row_list is None: 853 return {} 854 if len(row_list) == 0: 855 return {} 856 return dict (row_list)
857 #----------------------------------------------------------------
858 -def getMaritalStatusTypes():
859 """Gets a dictionary matching marital status types to their internal ID""" 860 row_list = gmPG.run_ro_query('personalia', "select name, pk from dem.marital_status") 861 if row_list is None: 862 return {} 863 if len(row_list) == 0: 864 return {} 865 return dict(row_list)
866 #------------------------------------------------------------------
867 -def getRelationshipTypes():
868 """Gets a dictionary of relationship types to internal id""" 869 row_list = gmPG.run_ro_query('personalia', "select description, id from dem.relation_types") 870 if row_list is None: 871 return None 872 if len (row_list) == 0: 873 return None 874 return dict(row_list)
875 876 #----------------------------------------------------------------
877 -def getUrb (id_urb):
878 cmd = """ 879 select 880 dem.state.name, 881 dem.urb.postcode 882 from 883 dem.urb, 884 dem.state 885 where 886 dem.urb.id = %s and 887 dem.urb.id_state = dem.state.id""" 888 row_list = gmPG.run_ro_query('personalia', cmd, None, id_urb) 889 if not row_list: 890 return None 891 else: 892 return (row_list[0][0], row_list[0][1])
893
894 -def getStreet (id_street):
895 cmd = """ 896 select 897 dem.state.name, 898 coalesce (dem.street.postcode, dem.urb.postcode), 899 dem.urb.name 900 from 901 dem.urb, 902 dem.state, 903 dem.street 904 where 905 dem.street.id = %s and 906 dem.street.id_urb = dem.urb.id and 907 dem.urb.id_state = dem.state.id 908 """ 909 row_list = gmPG.run_ro_query('personalia', cmd, None, id_street) 910 if not row_list: 911 return None 912 else: 913 return (row_list[0][0], row_list[0][1], row_list[0][2])
914
915 -def getCountry (country_code):
916 row_list = gmPG.run_ro_query('personalia', "select name from dem.country where code = %s", None, country_code) 917 if not row_list: 918 return None 919 else: 920 return row_list[0][0]
921 #-------------------------------------------------------------------------------
922 -def get_town_data (town):
923 row_list = gmPG.run_ro_query ('personalia', """ 924 select 925 dem.urb.postcode, 926 dem.state.code, 927 dem.state.name, 928 dem.country.code, 929 dem.country.name 930 from 931 dem.urb, 932 dem.state, 933 dem.country 934 where 935 dem.urb.name = %s and 936 dem.urb.id_state = dem.state.id and 937 dem.state.country = dem.country.code""", None, town) 938 if not row_list: 939 return (None, None, None, None, None) 940 else: 941 return tuple (row_list[0])
942 #============================================================ 943 # callbacks 944 #------------------------------------------------------------
945 -def _post_patient_selection(**kwargs):
946 print "received post_patient_selection notification" 947 print kwargs['kwds']
948 #============================================================ 949 950 #============================================================ 951 # main 952 #------------------------------------------------------------ 953 if __name__ == "__main__": 954 955 if len(sys.argv) < 2: 956 sys.exit() 957 958 if sys.argv[1] != 'test': 959 sys.exit() 960 961 import random 962 #--------------------------------------------------------
963 - def test_address_exists():
964 965 addresses = [ 966 { 967 'country': 'Germany', 968 'state': 'Sachsen', 969 'urb': 'Leipzig', 970 'postcode': '04318', 971 'street': u'Cunnersdorfer Strasse', 972 'number': '11' 973 }, 974 { 975 'country': 'DE', 976 'state': 'SN', 977 'urb': 'Leipzig', 978 'postcode': '04317', 979 'street': u'Riebeckstraße', 980 'number': '65', 981 'subunit': 'Parterre' 982 }, 983 { 984 'country': 'DE', 985 'state': 'SN', 986 'urb': 'Leipzig', 987 'postcode': '04317', 988 'street': u'Riebeckstraße', 989 'number': '65', 990 'subunit': '1. Stock' 991 }, 992 { 993 'country': 'DE', 994 'state': 'SN', 995 'urb': 'Leipzig', 996 'postcode': '04317', 997 'street': u'Riebeckstraße', 998 'number': '65', 999 'subunit': '1. Stock' 1000 }, 1001 { 1002 # 'country': 'DE', 1003 # 'state': 'SN', 1004 'urb': 'Leipzig', 1005 'postcode': '04317', 1006 'street': u'Riebeckstraße', 1007 'number': '65', 1008 'subunit': '1. Stock' 1009 }, 1010 ] 1011 1012 for adr in addresses: 1013 print adr 1014 exists = address_exists(**adr) 1015 if exists is None: 1016 print "address does not exist" 1017 else: 1018 print "address exists, primary key:", exists
1019 1020 #--------------------------------------------------------
1021 - def test_create_address():
1022 address = create_address ( 1023 country ='DE', 1024 state ='SN', 1025 urb ='Leipzig', 1026 suburb ='Sellerhausen', 1027 postcode ='04318', 1028 street = u'Cunnersdorfer Strasse', 1029 number = '11' 1030 # ,notes_subunit = '4.Stock rechts' 1031 ) 1032 print "created existing address" 1033 print address.format() 1034 1035 su = str(random.random()) 1036 1037 address = create_address ( 1038 country ='DE', 1039 state = 'SN', 1040 urb ='Leipzig', 1041 suburb ='Sellerhausen', 1042 postcode ='04318', 1043 street = u'Cunnersdorfer Strasse', 1044 number = '11', 1045 # notes_subunit = '4.Stock rechts', 1046 subunit = su 1047 ) 1048 print "created new address with subunit", su 1049 print address 1050 print address.format() 1051 print "deleted address:", delete_address(pk_address = address['pk_address'])
1052 #--------------------------------------------------------
1053 - def test_get_countries():
1054 for c in get_countries(): 1055 print c
1056 #--------------------------------------------------------
1057 - def test_get_country_for_region():
1058 region = raw_input("Please enter a region: ") 1059 print "country for region [%s] is: %s" % (region, get_country_for_region(region = region))
1060 #--------------------------------------------------------
1061 - def test_delete_tag():
1062 if delete_tag_image(tag_image = 9999): 1063 print "deleted tag 9999" 1064 else: 1065 print "did not delete tag 9999" 1066 if delete_tag_image(tag_image = 1): 1067 print "deleted tag 1" 1068 else: 1069 print "did not delete tag 1"
1070 #--------------------------------------------------------
1071 - def test_tag_images():
1072 tag = cTagImage(aPK_obj = 1) 1073 print tag
1074 #print get_tag_images() 1075 #--------------------------------------------------------
1076 - def test_get_billing_address():
1077 print get_patient_address_by_type(pk_patient = 12, adr_type = u'billing')
1078 #-------------------------------------------------------- 1079 #gmPG2.get_connection() 1080 1081 #test_address_exists() 1082 #test_create_address() 1083 #test_get_countries() 1084 #test_get_country_for_region() 1085 #test_delete_tag() 1086 test_tag_images() 1087 #test_get_billing_address() 1088 1089 sys.exit() 1090 #============================================================ 1091