from taskmaster.db import db
for key in db.keys('user-emails>*'):
email = key[len('user-emails>'):]
db.rename(key, 'email-lookup-user>%s' % email)
for key in db.keys('org-names>*'):
orgname = key[len('org-names>'):]
db.rename(key, 'name-lookup-org>%s' % orgname)
def delete_old_key(prefix):
for key in db.keys(prefix + '*'):
val = key[len(prefix):]
if '>' in val:
continue
try:
int(val, 16)
if len(val) != 32:
db.delete(key)
except ValueError:
db.delete(key)
delete_old_key('org>')
delete_old_key('queue>')
delete_old_key('user>')
delete_old_key('user>orgs>')
# Set all orgs to private
for org_key in db.keys('org>*'):
db.hset(org_key, 'level', 'private')
for key in db.keys('org-admins>*'):
org_id = key[len('org-admins>'):]
for user_id in db.smembers(key):
db.set('user-role>%s>%s' % (user_id, org_id), 'owner')
db.rename(key, 'org-users>%s' % org_id)
for queue_name in org_queue_names:
try:
# Check it isn't already converted
int(queue_name, 16)
continue
except ValueError:
pass
queue = queue_model.create({
'name': queue_name,
'org': org['id'],
})
# Have tasks reference queue id instead of queue name
try:
db.rename('queue-tasks2>%s' % queue_name,
'queue-tasks2>%s' % queue['id'])
except:
print "Queue rename failed on %s" % queue_name
for task_id in db.zrange('queue-tasks2>%s' % queue['id'], 0, -1):
task_model.update(task_id, 'queue', queue['id'], internal=True)
# Have tasks reference org id instead of org name
try:
db.rename('org-tasks2>%s' % orgname, 'org-tasks2>%s' % org['id'])
except:
print "Org rename failed on %s" % orgname
for task_id in db.zrange('org-tasks2>%s' % org['id'], 0, -1):
task_model.update(task_id, 'org', org['id'], internal=True)
task = task_model.get(task_id)
for queue_name in org_queue_names:
try:
# Check it isn't already converted
int(queue_name, 16)
continue
except ValueError:
pass
queue = queue_model.create({
'name': queue_name,
'org': org['id'],
})
# Have tasks reference queue id instead of queue name
try:
db.rename('queue-tasks2>%s' % queue_name, 'queue-tasks2>%s' % queue['id'])
except:
print "Queue rename failed on %s" % queue_name
for task_id in db.zrange('queue-tasks2>%s' % queue['id'], 0, -1):
task_model.update(task_id, 'queue', queue['id'], internal=True)
# Have tasks reference org id instead of org name
try:
db.rename('org-tasks2>%s' % orgname, 'org-tasks2>%s' % org['id'])
except:
print "Org rename failed on %s" % orgname
for task_id in db.zrange('org-tasks2>%s' % org['id'], 0, -1):
task_model.update(task_id, 'org', org['id'], internal=True)
task = task_model.get(task_id)
