def make_storage_user(user_id, username, visible_name, max_storage_bytes,
shard_id=None):
"""Create or update a StorageUser."""
gw = SystemGateway()
if not shard_id:
shard_id = get_new_user_shard_id(user_id)
return gw.create_or_update_user(
user_id, username, visible_name, max_storage_bytes, shard_id=shard_id)
def make_storage_user(user_id, username, visible_name, max_storage_bytes, shard_id=None):
"""Create or update a StorageUser."""
gw = SystemGateway()
if not shard_id:
shard_id = get_new_user_shard_id(user_id)
return gw.create_or_update_user(user_id, username, visible_name, max_storage_bytes, shard_id=shard_id)
(options, args) = parser.parse_args()
from utilities import userutils
import uuid
from backends.filesync.data.services import make_storage_user
from backends.filesync.data.dbmanager import get_new_user_shard_id
token_data = {}
for i in range(int(options.count)):
username = u"testuser%s" % i
userinfo = {
'username': unicode(uuid.uuid4()),
'full_name': u"name %s" % i,
'active': True,
'email': u"*****@*****.**",
}
# create the user account
user = userutils.create_user(userinfo)
# create the storage account
make_storage_user(
user.id, user.username,
"%s %s" % (user.first_name, user.last_name), 2 * (2 ** 30),
shard_id=get_new_user_shard_id(user.id))
# get an oauth token
token = userutils.make_oauth_token(user)
token_data[username] = (user.id, str(token))
token_file = os.path.join('testoauthkeys.json')
with open(token_file, 'w') as f:
json.dump(token_data, f)
