scripts/export-omf.py

   1 #!/usr/bin/plcsh
   2
   3 # utility to store, read and diff our user base data for exporting to OMF
   4
   5 import sys
   6 import time
   7 from optparse import OptionParser
   8
   9 import PLC.Persons
  10 import PLC.Methods.GetPersons
  11
  12 def getPersons(filter={}):
  13     m = PLC.Methods.GetPersons
  14
  15     m.hidden_fields.remove('password')
  16     return_fields = PLC.Persons.Person.fields.items()
  17     return_fields = dict([f for f in return_fields if f[0] not in m.hidden_fields])
  18
  19     return m.GetPersons(api).call(api, filter, return_fields)
  20
  21 class OmfUserBase:
  22
  23     def __init__ (self,filename):
  24         self.filename = filename
  25
  26     def save_person (self, file, person, sites_by_id, pubkeys_by_id):
  27         # do not expose people without a key
  28         if not person['key_ids']: return
  29         # do not expose techs
  30         roles=person['roles']
  31         if 'admin' not in roles and 'pi' not in roles and 'user' not in roles: return
  32         print >>file, "--- PERSON ---"
  33         #print >>file, "TMP id=%d"%person['person_id']
  34         print >>file, "email=%s"%person['email']
  35         for site_id in person['site_ids']:
  36             print >>file, "site=%s"%sites_by_id[site_id]['name']
  37         if 'pi' in person['roles']:
  38             print >>file, "pi=yes"
  39         print >>file, "password=%s"%person['password']
  40         for key_id in person['key_ids']:
  41             print >>file, "ssh=%s"%pubkeys_by_id[key_id]
  42
  43     def save(self):
  44         """
  45         Write configuration store to file.
  46         """
  47
  48         fileout = open(self.filename, 'w')
  49         print >>fileout, "# myplc simple userbase extration tools export-omf.py"
  50         print >>fileout, "# generated on %s"%time.strftime("%Y-%m-%d @ %H:%M UTC",time.gmtime())
  51         # store all ssh keys by key_id - keep only the public key
  52         pubkeys_by_id = dict ( [ (k['key_id'], k['key'].strip()) for k in GetKeys() if k['key_type'] == 'ssh'])
  53         # idem for sites, keep whole site info
  54         sites_by_id = dict ( [ (s['site_id'], s ) for s in GetSites({'peer_id':None}) ] )
  55         #
  56         persons=getPersons({'peer_id':None})
  57         persons = sorted(persons, key=lambda(person): person['email'])
  58         for p in persons:
  59             self.save_person(fileout, p, sites_by_id, pubkeys_by_id)
  60         fileout.close()
  61
  62
  63 def main ():
  64     output=sys.argv[1]
  65     userbase=OmfUserBase(output)
  66     userbase.save()
  67
  68 if __name__ == '__main__':
  69     main()