mirror of
https://github.com/sartography/uva-covid19-testing-communicator.git
synced 2025-02-24 12:58:05 +00:00
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import re
|
|
|
|
from flask import request
|
|
|
|
from communicator import app
|
|
from communicator.errors import CommError
|
|
from communicator.models.user import User
|
|
|
|
|
|
class UserService(object):
|
|
"""Provides details about the current User as read in from Shibboleth headers """
|
|
# ----------------------------------------
|
|
# Shibboleth Authentication Headers
|
|
# ----------------------------------------
|
|
# X-Remote-Cn: Daniel Harold Funk (dhf8r)
|
|
# X-Remote-Sn: Funk
|
|
# X-Remote-Givenname: Daniel
|
|
# X-Remote-Uid: dhf8r
|
|
# Eppn: dhf8r@virginia.edu
|
|
# Cn: Daniel Harold Funk (dhf8r)
|
|
# Sn: Funk
|
|
# Givenname: Daniel
|
|
# Uid: dhf8r
|
|
# X-Remote-User: dhf8r@virginia.edu
|
|
# X-Forwarded-For: 128.143.0.10
|
|
# X-Forwarded-Host: dev.crconnect.uvadcos.io
|
|
# X-Forwarded-Server: dev.crconnect.uvadcos.io
|
|
# Connection: Keep-Alive
|
|
|
|
def get_user_info(self):
|
|
if app.config['PRODUCTION']:
|
|
uid = request.headers.get("Uid")
|
|
cn = request.headers.get("Cn")
|
|
if not uid:
|
|
uid = request.headers.get("X-Remote-Uid")
|
|
if not uid:
|
|
raise CommError(1100, "invalid_sso_credentials", r"'Uid' nor 'X-Remote-Uid' were present in the headers: %s"% str(request.headers))
|
|
return User(uid, cn)
|
|
else:
|
|
return User('testUser', "Test User")
|
|
|
|
def is_valid_user(self):
|
|
user = self.get_user_info()
|
|
valid_ids = [x for x in re.compile('\s*[,|\s+]\s*').split(app.config['ADMINS'])]
|
|
return user.uid in valid_ids
|
|
|