plstackapi/importer/plclassic/site_importer.py

   1 from plstackapi.core.models import Site
   2
   3 class SiteImporter:
   4
   5     def __init__(self, api):
   6         self.api = api
   7         self.remote_sites = {}
   8         self.local_sites = {}
   9
  10     def run(self):
  11         db_sites = Site.objects.all()
  12         for db_site in db_sites:
  13             self.local_sites[db_site.login_base] = db_site
  14
  15         sites = self.api.GetSites()
  16         count = 0
  17         for site in sites:
  18             self.remote_sites[site['site_id']] = site
  19             if site['login_base'] not in self.local_sites:
  20                 new_site = Site(name=site['name'],
  21                                 site_url=site['url'],
  22                                 enabled=site['enabled'],
  23                                 longitude=site['longitude'],
  24                                 latitude=site['latitude'],
  25                                 is_public=site['is_public'],
  26                                 abbreviated_name=site['abbreviated_name'])
  27                 new_site.save()
  28                 count += 1
  29                 self.local_sites[new_site.login_base] = new_site
  30         print "imported %s sites" % count