Revision 50149cc8
| b/snf-pithos-backend/pithos/backends/lib/sqlalchemy/alembic/versions/165ba3fbfe53_update_path_account.py | ||
|---|---|---|
| 11 | 11 |
down_revision = '3dd56e750a3' |
| 12 | 12 |
|
| 13 | 13 |
from alembic import op |
| 14 |
from sqlalchemy.sql import table, column |
|
| 14 |
from sqlalchemy.sql import table, column, literal, and_
|
|
| 15 | 15 |
|
| 16 | 16 |
from synnefo.lib.astakos import get_user_uuid, get_username as get_user_username |
| 17 |
from pithos.api.settings import SERVICE_TOKEN, USER_INFO_URL |
|
| 17 |
from pithos.api.settings import ( |
|
| 18 |
SERVICE_TOKEN, USER_INFO_URL, AUTHENTICATION_USERS) |
|
| 18 | 19 |
|
| 19 | 20 |
import sqlalchemy as sa |
| 20 | 21 |
|
| ... | ... | |
| 25 | 26 |
if uuid: |
| 26 | 27 |
return uuid |
| 27 | 28 |
try: |
| 28 |
uuid = get_user_uuid(SERVICE_TOKEN, account, USER_INFO_URL) |
|
| 29 |
uuid = get_user_uuid( |
|
| 30 |
SERVICE_TOKEN, account, USER_INFO_URL, AUTHENTICATION_USERS) |
|
| 29 | 31 |
except Exception, e: |
| 30 | 32 |
print 'Unable to retrieve uuid for %s: %s' % (account, e) |
| 31 | 33 |
return |
| ... | ... | |
| 41 | 43 |
if username: |
| 42 | 44 |
return username |
| 43 | 45 |
try: |
| 44 |
username = get_user_username(SERVICE_TOKEN, account, USER_INFO_URL) |
|
| 46 |
username = get_user_username( |
|
| 47 |
SERVICE_TOKEN, account, USER_INFO_URL, AUTHENTICATION_USERS) |
|
| 45 | 48 |
except Exception, e: |
| 46 | 49 |
print 'Unable to retrieve username for %s: %s' % (account, e) |
| 47 | 50 |
return |
| ... | ... | |
| 68 | 71 |
column('path', sa.String(2048))
|
| 69 | 72 |
) |
| 70 | 73 |
|
| 74 |
xvals = table( |
|
| 75 |
'xfeaturevals', |
|
| 76 |
column('feature_id', sa.Integer),
|
|
| 77 |
column('key', sa.Integer),
|
|
| 78 |
column('value', sa.String(256))
|
|
| 79 |
) |
|
| 80 |
|
|
| 81 |
|
|
| 71 | 82 |
def upgrade(): |
| 72 | 83 |
connection = op.get_bind() |
| 73 | 84 |
|
| ... | ... | |
| 104 | 115 |
u = x.update().where(x.c.feature_id == id).values({'path':path})
|
| 105 | 116 |
connection.execute(u) |
| 106 | 117 |
|
| 118 |
s = sa.select([xvals.c.feature_id, xvals.c.key, xvals.c.value]) |
|
| 119 |
s = s.where(xvals.c.value != '*') |
|
| 120 |
xfeaturevals = connection.execute(s).fetchall() |
|
| 121 |
for feature_id, key, value in xfeaturevals: |
|
| 122 |
account, sep, group = value.partition(':')
|
|
| 123 |
uuid = get_uuid(account) |
|
| 124 |
if not uuid: |
|
| 125 |
continue |
|
| 126 |
new_value = sep.join([uuid, group]) |
|
| 127 |
u = xvals.update() |
|
| 128 |
u = u.where(and_( |
|
| 129 |
xvals.c.feature_id == feature_id, |
|
| 130 |
xvals.c.key == key, |
|
| 131 |
xvals.c.value == value)) |
|
| 132 |
u = u.values({'value':new_value})
|
|
| 133 |
connection.execute(u) |
|
| 134 |
|
|
| 107 | 135 |
|
| 108 | 136 |
def downgrade(): |
| 109 | 137 |
connection = op.get_bind() |
| ... | ... | |
| 140 | 168 |
path = sep.join([username, rest]) |
| 141 | 169 |
u = x.update().where(x.c.feature_id == id).values({'path':path})
|
| 142 | 170 |
connection.execute(u) |
| 171 |
|
|
| 172 |
s = sa.select([xvals.c.feature_id, xvals.c.key, xvals.c.value]) |
|
| 173 |
s = s.where(xvals.c.value != '*') |
|
| 174 |
xfeaturevals = connection.execute(s).fetchall() |
|
| 175 |
for feature_id, key, value in xfeaturevals: |
|
| 176 |
account, sep, group = value.partition(':')
|
|
| 177 |
username = get_username(account) |
|
| 178 |
if not username: |
|
| 179 |
continue |
|
| 180 |
new_value = sep.join([username, group]) |
|
| 181 |
u = xvals.update() |
|
| 182 |
u = u.where(and_( |
|
| 183 |
xvals.c.feature_id == feature_id, |
|
| 184 |
xvals.c.key == key, |
|
| 185 |
xvals.c.value ==value)) |
|
| 186 |
u = u.values({'value':new_value})
|
|
| 187 |
connection.execute(u) |
|
Also available in: Unified diff