projects.py: fix visibility import

This commit is contained in:
Drew DeVault 2020-04-29 10:07:47 -04:00
parent a15ff75966
commit b5af713dcc
1 changed files with 3 additions and 4 deletions

View File

@ -1,5 +1,5 @@
from flask import abort
from hubsrht.types import Project, User
from hubsrht.types import Project, User, Visibility
from srht.oauth import current_user
from enum import Enum
@ -21,9 +21,8 @@ def get_project(owner, project_name, access, user=current_user):
if access == ProjectAccess.write:
abort(401)
# TODO: ACLs
if project.visibility in (ProjectVisibility.public,
ProjectVisibility.unlisted):
if project.visibility in (Visibility.public, Visibility.unlisted):
return project.owner, project
elif project.visibility == ProjectVisibility.private:
elif project.visibility == Visibility.private:
abort(401)
assert False