python3 tags are gathered in a tags file
[infrastructure.git] / migration_scripts / Migration.py
1 #!/usr/bin/env python
2 #this script is for migrating sites, user from PLC to PLE
3 import os,sys, time
4 import xmlrpclib
5 from optparse import OptionParser
6 from pprint import PrettyPrinter
7
8
9
10 class Migration:
11     subversion_id = "$Id Amine"
12
13     def __init__(self):
14         try:
15         #Remote(PLC) Api connexion
16             self.remote_name="PLC"
17             self.remote_url="https://tac.inria.fr:443/PLCAPI/"
18             self.remote_server=xmlrpclib.Server(self.remote_url,allow_none=True)
19         #local(PLE) Api connexion
20             self.local_name="PLE"
21             self.local_url="https://garfield.inria.fr:443/PLCAPI/"
22             self.local_server=xmlrpclib.Server(self.local_url,allow_none=True)
23         except : raise
24         
25     def local_auth(self):
26         return {'Username': 'root@plc2.inria.fr',
27                 'AuthMethod':'password',
28                 'AuthString':'*****',
29                 'Role' : 'root'
30                 }
31
32     def remote_auth(self):
33         return {'Username': 'root@plc1.inria.fr',
34                 'AuthMethod':'password',
35                 'AuthString':'******',
36                 'Role' : 'root'
37                 }
38     
39     def header(self,message):
40         now=time.strftime("%H:%M:%S", time.localtime())
41         print "*",now,'--',message
42
43     def migrate_persons(self, person_ids, site_id):
44         self.header("Dealing with persons migration")
45         for person_id in person_ids:
46             remote_person=self.remote_server.GetPersons(self.remote_auth(),{'person_id':person_id})
47             remote_person[0]['peer_id']= None
48             self.header("Migrating from %s to %s the person %s \n"%(self.remote_name,self.local_name,
49                                                                     remote_person[0]))
50             local_person=self.local_server.AddPerson(self.local_auth(),remote_person[0])
51             self.header("Add %s to Site %d \n"%(remote_person[0]['email'],site_id))
52             status=self.local_server.AddPersonToSite(self.local_auth(),
53                                                      remote_person[0]['email'],site_id)
54             #to be coded
55                #Add roles to person the same that he has in his remote platform
56                #Adding keys
57                #Adding person to slice
58                
59     def migrate_sites(self,sites_names):
60         self.header("Dealing with sites migration")
61         for site_name in sites_names:
62             remote_site=self.remote_server.GetSites(self.remote_auth(),{'name':site_name})
63             self.header("Migrating from %s the site %s \n"%(self.remote_name,remote_site))
64             remote_persons_ids=remote_site[0]['person_ids']
65             peer_site_id=remote_site[0]['site_id']
66             #make some changes to the new incoming site
67             remote_site[0]['peer_site_id']=peer_site_id
68             remote_site[0]['peer_id']= None
69             remote_site[0]['login_base']=remote_site[0]['login_base']+'ple'
70             remote_site[0]['enabled']=True
71             local_site=self.local_server.AddSite(self.local_auth(),remote_site[0])
72             #May be we have to manage the site address too.... to be coded
73             self.header("Getting in %s a new  site %s\n"%(self.local_name,
74                                                           self.local_server.GetSites(self.local_auth(),
75                                                                                      {'site_id':local_site})))
76             #getting person involved in the remote site
77             self.migrate_persons(remote_persons_ids,local_site)
78             return True
79         
80     def run (self):
81         usage = """usage: %%prog  sites_names [site_name1][site_name2].......
82         """
83         parser=OptionParser(usage=usage,version=self.subversion_id)
84         parser.add_option("-v","--verbose", action="store_true", dest="verbose", default=False, 
85                           help="Run in verbose mode")
86         (self.options, self.args) = parser.parse_args()
87         
88         if len(self.args) == 0:
89             print usage
90         else:
91             self.migrate_sites(self.args)
92         
93     
94     def main(self):
95         try:
96             success=self.run()
97             if success:
98                 return 0
99             else:
100                 return 1 
101         except SystemExit:
102             raise
103   
104 if __name__ == "__main__":
105     sys.exit(Migration().main())