blob: d513fb67118588c910d95b329c8eefe1ff431a29 [file] [log] [blame]
Tony Mack743ddf22013-04-02 07:36:25 -04001from plstackapi.core.models import Site
Tony Mackf593dba2013-03-24 13:17:43 -04002
3class SiteImporter:
4
Tony Mack743ddf22013-04-02 07:36:25 -04005 def __init__(self, api):
6 self.api = api
7 self.remote_sites = {}
8 self.local_sites = {}
Tony Mackf593dba2013-03-24 13:17:43 -04009
10 def run(self):
Tony Mackf593dba2013-03-24 13:17:43 -040011 db_sites = Site.objects.all()
Tony Mack743ddf22013-04-02 07:36:25 -040012 for db_site in db_sites:
13 self.local_sites[db_site.login_base] = db_site
Tony Mackdd12c682013-04-02 10:15:27 -040014 print "%s local sites" % len(db_sites)
Tony Mack743ddf22013-04-02 07:36:25 -040015
16 sites = self.api.GetSites()
Tony Mackd9f8d2e2013-04-02 10:18:48 -040017 print "%s remote sites" % len(sites)
Tony Mackf1afe392013-04-02 07:40:38 -040018 count = 0
Tony Mackf593dba2013-03-24 13:17:43 -040019 for site in sites:
Tony Mack743ddf22013-04-02 07:36:25 -040020 self.remote_sites[site['site_id']] = site
21 if site['login_base'] not in self.local_sites:
Tony Mackf593dba2013-03-24 13:17:43 -040022 new_site = Site(name=site['name'],
23 site_url=site['url'],
24 enabled=site['enabled'],
25 longitude=site['longitude'],
26 latitude=site['latitude'],
27 is_public=site['is_public'],
28 abbreviated_name=site['abbreviated_name'])
29 new_site.save()
Tony Mackf1afe392013-04-02 07:40:38 -040030 count += 1
Tony Mack743ddf22013-04-02 07:36:25 -040031 self.local_sites[new_site.login_base] = new_site
Tony Mackf1afe392013-04-02 07:40:38 -040032 print "imported %s sites" % count