mandaye_vincennes_manager: fix migration

This commit is contained in:
Jérôme Schneider 2014-04-16 19:29:12 +02:00
parent e21ea8408b
commit 7c2d54f0ca
1 changed files with 14 additions and 9 deletions

View File

@ -1,4 +1,4 @@
#!/usr/bin/env python
#! /usr/bin/python
# -*- coding: utf-8 -*-
""" Script to administrate mandaye server
@ -77,22 +77,24 @@ def main():
for user in ManagerSPUser.all():
user.password = encrypt_pwd(user.password)
ManagerSPUser.save()
if options.db_old:
if options.old_db:
from mandaye.db import sql_session
from mandaye.models import IDPUser, SPUser, ServiceProvider
from sqlalchemy import create_engine
from sqlalchemy.sql import text
engine = create_engine(options.old_db)
conn = engine.engine()
conn = engine.connect()
sites = conn.execute(text("select id, name FROM sites")).fetchall()
for site in sites:
sp = ServiceProvider(id=site[0], name=name[0])
sp = ServiceProvider(name=site[1])
sp.id = site[0]
sql_session().add(sp)
sql_session().commit()
local_users = conn.execute(text("select id, login from local_users")).fetchall()
for local_user in local_users:
idp_user = IDPUser(id=local_user[0], unique_id=local_user[1], idp_id='default')
sql_session().add(local_user)
idp_user = IDPUser(unique_id=local_user[1], idp_id='default')
idp_user.id = local_user[0]
sql_session().add(idp_user)
sql_session().commit()
ext_users = conn.execute(text("select id, login, password, birthdate, last_connection, local_user_id, site_id, creation_date from ext_users")).fetchall()
for ext_user in ext_users:
@ -108,9 +110,12 @@ def main():
post_values['txtNomFoyer'] = ext_user[1]
post_values['txtCode'] = ext_user[2]
post_values['txtDateNaissance'] = ext_user[3]
sp_user = SPUser(id=ext_user[0], login=ext_user[1], post_values=post_values,
creation_date=ext_user[7], last_connection=ext_user[4],
idp_user_id=ext_user[5], service_provider_id=site_id)
sp_user = SPUser(login=ext_user[1], post_values=post_values)
sp_user.idp_user_id = ext_user[5]
sp_user.id = ext_user[0]
sp_user.last_connection = ext_user[4]
sp_user.service_provider_id = site_id
sp_user.creation_date = ext_user[7]
sql_session().add(sp_user)
sql_session().commit()