blob: 0db9136ecdf991e389461f6c47bb0118aff32f62 [file] [log] [blame]
from plstackapi.core.models import Slice
class SliceImporter:
def __init__(self, api):
self.api = api
self.remote_slices = {}
self.local_slices = {}
def run(self, remote_sites={}, local_sites={}):
if not remote_sites:
sites = self.api.GetSites()
for site in sites:
remote_sites[site['site_id']] = site
if not local_sites:
from models import Site
sites = Site.objects.all()
for site in sites:
local_sites[site.login_base] = site
db_slices = Slice.objects.all()
for db_slice in db_slices:
self.local_slices[db_slice.name] = db_slice
slices = api.GetSlices()
count = 0
for slice in slices:
self.remote_slice[slice['slice_id']] = slice
if slice['name'] not in self.local_slices:
site = local_sites[remote_sites[slice['site_id']]['login_base']]
new_slices = Slice(name=slice['name'],
instantiation=slice['instantiation'],
omf_friendly = False,
description = slice['description'],
slice_url = slice['url'],
site = site)
new_slice.save()
count += 1
self.local_slices[new_slice.name] = new_slice
print "Imported %s slices" % count