def post(self): key = self.request.get('key') email = self.request.get('email').strip() user = User.get(key) user.email = email first_name = self.request.get('first_name').strip() user.first_name = first_name last_name = self.request.get('last_name').strip() user.last_name = last_name position = self.request.get('position') user.position = position first_date = self.request.get('first_date') try: first_date = datetime.datetime.strptime(first_date, '%Y-%m-%d').date() except ValueError: first_date = None user.first_date = first_date dept = self.request.get('dept') dept_ref = Dept.all().filter('name', dept).get() if dept_ref is None: dept_ref = Dept(name=dept) dept_ref.put() user.dept = dept_ref try: manager = Model.get(self.request.get('manager')) except BadKeyError: manager = None user.manager = manager roles = self.request.get('role')[:-1].split(',') user.role = [] for role in roles: role_key = Role.gql("WHERE value = :role", role=role).get().key() user.role.append(role_key) user.put() self.response.out.write('You have successfully updated user info')
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')