def upgrade(upd, with_testing, *args, **kwargs):
upd.print_log('Upgrading db...')
helpers.upgrade_db(revision='3d2db4a87f86')
upd.print_log('Make HostingPanel role internal...')
role = Role.filter(Role.rolename == 'HostingPanel').one()
role.internal = True
role.save()
def upgrade(upd, with_testing, *args, **kwargs):
upd.print_log('Add menus Persistent volumes and Public IPs')
user = Role.filter(Role.rolename == "User").one()
trial_user = Role.filter(Role.rolename == "TrialUser").one()
public_ips = MenuItem.create(name="Public IPs",
path="/publicIPs/",
ordering=1,
menu_id=1)
public_ips.save()
perm = MenuItemRole(role=user, menuitem=public_ips)
perm = MenuItemRole(role=trial_user, menuitem=public_ips)
perm.save()
p = MenuItem.create(name="Persistent volumes",
path="/persistent-volumes/",
ordering=2,
menu_id=1)
p.save()
perm = MenuItemRole(role=user, menuitem=p)
perm = MenuItemRole(role=trial_user, menuitem=p)
perm.save()
def add_menu_items(items, menu, parent=None,):
for item in items:
roles = item.pop('roles', [])
children = item.pop('children', None)
menu_item = MenuItem(**item)
menu_item.menu = menu
menu_item.parent = parent
menu_item.save()
for rolename in roles:
role = Role.filter(Role.rolename == rolename).one()
item_role = MenuItemRole(role=role, menuitem=menu_item)
item_role.save()
if children:
add_menu_items(children, menu, parent=menu_item)
def upgrade(upd, with_testing, *args, **kwargs):
upd.print_log('Add MenuItemRole model.')
upd.print_log('Upgrading db...')
helpers.upgrade_db(revision='241a7b04a9ff')
admin = Role.filter(Role.rolename == 'Admin').first()
trialuser = Role.filter(Role.rolename == 'TrialUser').first()
user = Role.filter(Role.rolename == 'User').first()
menu_pods = MenuItem.filter(MenuItem.name == 'Pods').first()
menu_publicip = MenuItem.filter(MenuItem.name == 'Public IPs').first()
menu_pv = MenuItem.filter(MenuItem.name == 'Persistent volumes').first()
menu_nodes = MenuItem.filter(MenuItem.name == 'Nodes').first()
menu_papps = MenuItem.filter(
MenuItem.name == 'Predefined Applications').first()
menu_settings = MenuItem.filter(MenuItem.name == 'Settings').first()
menu_adm = MenuItem.filter(MenuItem.name == 'Administration').first()
menu_users = MenuItem.filter(MenuItem.name == 'Users').first()
menu_ippool = MenuItem.filter(MenuItem.name == 'IP pool').first()
MenuItemRole.create(menuitem=menu_pods, role=user)
MenuItemRole.create(menuitem=menu_pods, role=trialuser)
MenuItemRole.create(menuitem=menu_publicip, role=user)
MenuItemRole.create(menuitem=menu_publicip, role=trialuser)
MenuItemRole.create(menuitem=menu_pv, role=user)
MenuItemRole.create(menuitem=menu_pv, role=trialuser)
MenuItemRole.create(menuitem=menu_nodes, role=admin)
MenuItemRole.create(menuitem=menu_papps, role=admin)
MenuItemRole.create(menuitem=menu_settings, role=admin)
MenuItemRole.create(menuitem=menu_settings, role=user)
MenuItemRole.create(menuitem=menu_settings, role=trialuser)
MenuItemRole.create(menuitem=menu_adm, role=admin)
MenuItemRole.create(menuitem=menu_users, role=admin)
MenuItemRole.create(menuitem=menu_ippool, role=admin)
resource = Resource.filter(Resource.name == 'static_pages').first()
if resource:
Permission.filter(Permission.resource == resource).delete()
resource.delete()