class User(ndb.Model):
"""A single user of the IDE"""
# No parent
# key is the user's unique name
joinDate = ndb.DateTimeProperty(auto_now_add=True)
email = ndb.StringProperty()
gaeUser = ndb.UserProperty()
secret = ndb.StringProperty()
class SomeKind(ndb.Model):
user = ndb.UserProperty()
class Greeting(ndb.Model):
"""Models an individual Guestbook entry with author, content, and date."""
author = ndb.UserProperty()
content = ndb.StringProperty()
date = ndb.DateTimeProperty(auto_now_add=True)
class Employee(ndb.Model, Pagination):
"""Models an Employee."""
department = ndb.StringProperty()
first_name = ndb.StringProperty()
last_name = ndb.StringProperty()
meta_department = ndb.StringProperty()
photo_url = ndb.TextProperty()
terminated = ndb.BooleanProperty(default=False)
timestamp = ndb.DateTimeProperty(auto_now_add=True)
user = ndb.UserProperty()
username = ndb.StringProperty()
is_admin = ndb.BooleanProperty()
@classmethod
def get_current_employee(cls):
user = oidc.user_getfield('email').split('@')[0]
user_email = oidc.user_getfield('email')
employee = cls.query(cls.username == user,
cls.terminated == False).get() # noqa
if employee is None:
raise NoSuchEmployee(
'Couldn\'t find a Google Apps user with email {}'.format(
user_email))
return employee
@classmethod
def create_from_dict(cls, d, persist=True):
new_employee = cls()
new_employee.username = d['username']
new_employee.update_from_dict(d)
if persist is True:
new_employee.put()
return new_employee
@classmethod
def key_to_username(cls, key):
return cls.query(cls.key == key).get().username
@classmethod
@memoized
def get_key_for_username(cls, username):
key = cls.query(cls.username == username,
cls.terminated == False).get(keys_only=True) # noqa
if key is None:
raise NoSuchEmployee(
"Couldn't find a Google Apps user with username {}".format(
username))
return key
def update_from_dict(self, d):
self.first_name = d['first_name']
self.last_name = d['last_name']
self.photo_url = d.get('photo_url')
self.department = d.get('department')
self.meta_department = get_meta_department(self.department)
self.terminated = False
def is_employee_data_changed(self, d, is_terminated):
result = any([
self.first_name != d.get('first_name'),
self.last_name != d.get('last_name'),
self.photo_url != d.get('photo_url'),
self.department != d.get('department'),
self.meta_department != get_meta_department(self.department),
self.terminated is not is_terminated,
])
return result
def get_gravatar(self):
"""Creates gravatar URL from email address."""
m = hashlib.md5()
m.update(self.email.encode())
encoded_hash = base64.b16encode(m.digest()).lower()
return 'https://gravatar.com/avatar/{}?s=200'.format(encoded_hash)
def get_photo_url(self):
"""Return an avatar photo URL (depending on Gravatar config). This still could
be empty, in which case the theme needs to provide an alternate photo.
"""
if config.GRAVATAR == 'always':
return self.get_gravatar()
elif config.GRAVATAR == 'backup' and not self.photo_url:
return self.get_gravatar()
else:
return self.photo_url
@property
def full_name(self):
"""Return user's full name (first name + ' ' + last name)."""
return '{} {}'.format(self.first_name, self.last_name)
@property
def email(self):
"""Return user's email (username + '@' + config.Domain)."""
return '{}@{}'.format(self.username, config.DOMAIN)
