blob: 1802f6f147ed66f4f205e38acfb9840e6b52d2fa [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 Mackdd12c682013-04-02 10:15:27 -040017 print "%s remote sites" %s 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