fix app author
This commit is contained in:
parent
f87139ea24
commit
5e1e363b7d
|
@ -40,57 +40,52 @@ def get_shout_slug(entry):
|
||||||
def create_author_from_app(app):
|
def create_author_from_app(app):
|
||||||
user = None
|
user = None
|
||||||
userdata = None
|
userdata = None
|
||||||
with local_session() as session:
|
# check if email is used
|
||||||
# check if email is used
|
if app['email']:
|
||||||
user = session.query(User).where(User.email == app['email']).first()
|
with local_session() as session:
|
||||||
if not user:
|
user = session.query(User).where(User.email == app['email']).first()
|
||||||
# print('[migration] app %r' % app)
|
|
||||||
name = app.get('name')
|
|
||||||
if name:
|
|
||||||
slug = translit(name, "ru", reversed=True).lower()
|
|
||||||
slug = re.sub('[^0-9a-zA-Z]+', '-', slug)
|
|
||||||
# check if slug is used
|
|
||||||
if slug:
|
|
||||||
user = session.query(User).where(User.slug == slug).first()
|
|
||||||
|
|
||||||
# get slug from email
|
|
||||||
if user:
|
|
||||||
slug = app['email'].split('@')[0]
|
|
||||||
user = session.query(User).where(User.slug == slug).first()
|
|
||||||
# one more try
|
|
||||||
if user:
|
|
||||||
slug += '-author'
|
|
||||||
user = session.query(User).where(User.slug == slug).first()
|
|
||||||
else:
|
|
||||||
# print(f'[migration] author @{slug} is found by email')
|
|
||||||
pass
|
|
||||||
|
|
||||||
else:
|
|
||||||
# print(f'[migration] author @{slug} is found')
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
# create user with application data
|
|
||||||
if not user:
|
if not user:
|
||||||
userdata = {
|
# print('[migration] app %r' % app)
|
||||||
"username": app["email"],
|
name = app.get('name')
|
||||||
"email": app["email"],
|
if name:
|
||||||
"name": app.get("name", ""),
|
slug = translit(name, "ru", reversed=True).lower()
|
||||||
"bio": app.get("bio", ""),
|
slug = re.sub('[^0-9a-zA-Z]+', '-', slug)
|
||||||
"emailConfirmed": False,
|
print('[migration] created slug %s' % slug)
|
||||||
"slug": slug,
|
# check if slug is used
|
||||||
"createdAt": ts,
|
if slug:
|
||||||
"lastSeen": ts,
|
user = session.query(User).where(User.slug == slug).first()
|
||||||
}
|
|
||||||
# print('[migration] userdata %r' % userdata)
|
|
||||||
user = User.create(**userdata)
|
|
||||||
session.add(user)
|
|
||||||
session.commit()
|
|
||||||
userdata['id'] = user.id
|
|
||||||
|
|
||||||
if not userdata:
|
# get slug from email
|
||||||
userdata = user.dict()
|
if user:
|
||||||
|
slug = app['email'].split('@')[0]
|
||||||
|
user = session.query(User).where(User.slug == slug).first()
|
||||||
|
# one more try
|
||||||
|
if user:
|
||||||
|
slug += '-author'
|
||||||
|
user = session.query(User).where(User.slug == slug).first()
|
||||||
|
|
||||||
|
# create user with application data
|
||||||
|
if not user:
|
||||||
|
userdata = {
|
||||||
|
"username": app["email"],
|
||||||
|
"email": app["email"],
|
||||||
|
"name": app.get("name", ""),
|
||||||
|
"bio": app.get("bio", ""),
|
||||||
|
"emailConfirmed": False,
|
||||||
|
"slug": slug,
|
||||||
|
"createdAt": ts,
|
||||||
|
"lastSeen": ts,
|
||||||
|
}
|
||||||
|
# print('[migration] userdata %r' % userdata)
|
||||||
|
user = User.create(**userdata)
|
||||||
|
session.add(user)
|
||||||
|
session.commit()
|
||||||
|
userdata['id'] = user.id
|
||||||
|
|
||||||
|
userdata = user.dict()
|
||||||
return userdata
|
return userdata
|
||||||
|
else:
|
||||||
|
raise Exception("app is not ok", app)
|
||||||
|
|
||||||
|
|
||||||
async def create_shout(shout_dict):
|
async def create_shout(shout_dict):
|
||||||
|
@ -115,17 +110,14 @@ async def get_user(entry, storage):
|
||||||
user_oid = None
|
user_oid = None
|
||||||
if app:
|
if app:
|
||||||
userdata = create_author_from_app(app)
|
userdata = create_author_from_app(app)
|
||||||
# print("[migration] from app")
|
else:
|
||||||
|
|
||||||
if not userdata:
|
|
||||||
user_oid = entry.get("createdBy")
|
user_oid = entry.get("createdBy")
|
||||||
if user_oid == "0":
|
if user_oid == "0":
|
||||||
userdata = discours
|
userdata = discours
|
||||||
elif user_oid:
|
elif user_oid:
|
||||||
userdata = storage["users"]["by_oid"].get(user_oid)
|
userdata = storage["users"]["by_oid"].get(user_oid)
|
||||||
# print("[migration] user from user_oid")
|
if not userdata:
|
||||||
else:
|
print('no userdata by oid, anonymous')
|
||||||
# print("[migration] no app, no user_oid")
|
|
||||||
userdata = anondict
|
userdata = anondict
|
||||||
print(app)
|
print(app)
|
||||||
# cleanup slug
|
# cleanup slug
|
||||||
|
@ -141,7 +133,6 @@ async def get_user(entry, storage):
|
||||||
return user, user_oid
|
return user, user_oid
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def migrate(entry, storage):
|
async def migrate(entry, storage):
|
||||||
author, user_oid = await get_user(entry, storage)
|
author, user_oid = await get_user(entry, storage)
|
||||||
r = {
|
r = {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user