Skip to content
Snippets Groups Projects
  • Vangelis Koukis's avatar
    Switch license to GPLv3 · 02071b96
    Vangelis Koukis authored
    According to the decision of the GRNET Board of Directors,
    switch license to GPLv3.
    
    This commit will be propagated to the release
    and master branches based on git flow, and the next
    release will be licensed as GPLv3.
    02071b96
migrate-users 3.19 KiB
#!/usr/bin/env python

# Copyright (C) 2010-2014 GRNET S.A.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from sqlalchemy import Table
from sqlalchemy.sql import select

from pithos.im.models import User

from migrate import Migration

import base64

class UserMigration(Migration):
    def __init__(self, db):
        Migration.__init__(self, db)
        self.gss_users = Table('gss_user', self.metadata, autoload=True)
    
    def execute(self):
        for u in self.retrieve_users():
            user = User()
            user.pk = u['id']
            user.uniq = u['username']
            user.realname = u['name']
            user.affiliation = u['homeorganization'] if u['homeorganization'] else ''
            user.auth_token = base64.b64encode(u['authtoken'])
            user.auth_token_created = u['creationdate']
            user.auth_token_expires = u['authtokenexpirydate']
            user.created = u['creationdate']
            user.updated = u['modificationdate']
            user.email = u['email']
            user.active = 'ACTIVE' if u['active'] else 'SUSPENDED'
            print '#', user
            user.save(update_timestamps=False)
            
            #create user groups
            for (owner, group, members) in self.retrieve_groups(u['username']):
                self.backend.permissions.group_addmany(owner, group, members)
    
    
    def retrieve_users(self):
        s = self.gss_users.select()
        rp = self.conn.execute(s)
        user = rp.fetchone()
        while user:
            yield user
            user = rp.fetchone()
        rp.close()
    
    def retrieve_groups(self, owner):
        gss_group = Table('gss_group', self.metadata, autoload=True)
        gss_user = Table('gss_user', self.metadata, autoload=True)
        group_user = Table('gss_group_gss_user', self.metadata, autoload=True)
        j1 = gss_group.join(gss_user, gss_group.c.owner_id == gss_user.c.id)
        j2 = group_user.join(gss_user, group_user.c.members_id == gss_user.c.id)
        s = select([gss_group.c.id, gss_group.c.name, gss_user.c.username], from_obj=j1)
        s = s.where(gss_user.c.username == owner)
        rp = self.conn.execute(s)
        gr = rp.fetchone()
        while gr:
            id, group, owner = gr
            s = select([gss_user.c.username], from_obj=j2)
            s = s.where(group_user.c.groupsmember_id == id)
            rp2 = self.conn.execute(s)
            members = rp2.fetchall()
            rp2.close()
            yield owner, group, (m[0] for m in members)
            gr = rp.fetchone()
        rp.close()

if __name__ == "__main__":
    db = ''
    m = UserMigration(db)
    m.execute()