def __call__(self, environ, start_response):
current_user = users.get_current_user()
url = users.create_login_url("/")
req = Request(environ)
non_auth_urls = ['/create_role', '/p', '/users', '/add_emp', '/edit_user', '/upload_contacts', '/parse_xls']
if environ['PATH_INFO'] not in non_auth_urls:
if current_user is None:
try:
auth_header = req.headers['Authorization']
except KeyError:
resp = Response(status="307", location=url)
return resp(environ, start_response)
username, password = '', ''
try:
user_info = base64.decodestring(auth_header[6:])
username, password = user_info.split(':')
except ValueError:
resp = Response(status="401")
return resp(environ, start_response)
user_info = User.gql("WHERE username = :username ",
username=username).get()
if user_info is None:
resp = Response(status="401")
return resp(environ, start_response)
if not check_password(password, user_info.password):
resp = Response(status="401")
return resp(environ, start_response)
else:
email = str(current_user.email()).strip()
user_info = User.all().filter('email', email).get()
if user_info is None:
user_info = User(
email=current_user.email())
user_info.put()
environ["current_user"] = user_info
try:
environ["current_role"] = Model.get(user_info.role[0]).value
except IndexError:
environ["current_role"] = ''
resp = req.get_response(self.app)
return resp(environ, start_response)
def get(self):
file_key = ContactXlsFile.all().get().file_key
blob_info = blobstore.BlobInfo.get(file_key)
file = blob_info.open().read()
wb = xlrd.open_workbook(file_contents=file)
wb.sheet_names()
sh = wb.sheet_by_index(0)
cols_dict = {'first_name': 1,
'last_name': 0,
'dept': 2,
'position': 3,
'email': 4,
'manager': 12}
manager_dict = {}
reg = "^\s+|\n|\r|\s+$"
employee_role = Role.all().filter('value', 'employee').get().key()
for rownum in range(sh.nrows)[6:]:
email = sh.cell_value(rownum, cols_dict['email']).strip()
user = User.all().filter('email', email).get()
if user is None:
user = User(email=email)
user.put()
if not user.role:
user.role.append(employee_role)
string_fields = ['first_name',
'last_name',
'position',
]
for field in string_fields:
value = re.sub(reg,
'',
sh.cell_value(rownum, cols_dict[field]))
setattr(user, field, value)
department_str = re.sub(reg,
'',
sh.cell_value(rownum, cols_dict['dept']))
department = Dept.all().filter('name', department_str).get()
if department is None:
department = Dept(name=department_str)
department.put()
user.dept = department
user.put()
manager_str = re.sub(reg,
'',
sh.cell_value(rownum, cols_dict['manager']))
manager_dict[user.key()] = manager_str
manager_role = Role.all().filter('value', 'manager').get().key()
for user_key in manager_dict:
manager_str = manager_dict[user_key].replace(' ', ' ')
if manager_str:
last_name, first_name = manager_str.split(' ')[:2]
manager = User.all().filter('last_name',
last_name).filter('first_name',
first_name).get()
else:
manager = None
if manager is not None:
if manager_role not in manager.role:
manager.role.append(manager_role)
manager.put()
user = User.get(user_key)
user.manager = manager
user.put()
self.redirect('/users')
