migration, auth, refactoring, formatting

resolvers/collab.py

@@ -1,10 +1,11 @@
 from datetime import datetime
+
+from auth.authenticate import login_required
 from base.orm import local_session
+from base.resolvers import query, mutation
 from orm.collab import Collab
 from orm.shout import Shout
 from orm.user import User
-from base.resolvers import query, mutation
-from auth.authenticate import login_required
 
 
 @query.field("getCollabs")
@@ -12,11 +13,10 @@ from auth.authenticate import login_required
 async def get_collabs(_, info):
     auth = info.context["request"].auth
     user_id = auth.user_id
-    collabs = []
     with local_session() as session:
         user = session.query(User).where(User.id == user_id).first()
         collabs = session.query(Collab).filter(user.slug in Collab.authors)
-    return collabs
+        return collabs
 
 
 @mutation.field("inviteAuthor")
@@ -37,7 +37,7 @@ async def invite_author(_, info, author, shout):
             return {"error": "already added"}
         shout.authors.append(author)
         shout.updated_at = datetime.now()
-        shout.save()
+        session.add(shout)
         session.commit()
 
     # TODO: email notify
@@ -63,7 +63,7 @@ async def remove_author(_, info, author, shout):
             return {"error": "not in authors"}
         shout.authors.remove(author)
         shout.updated_at = datetime.now()
-        shout.save()
+        session.add(shout)
         session.commit()
 
     # result = Result("INVITED")