def add_staff_project(project_id, attendee_ids_str):
if attendee_ids_str not in (None, ''):
attendee_ids_list = attendee_ids_str.split(',')
for id in attendee_ids_list:
item = UserProject(user_id=id, project_id=project_id)
session.add(item)
session.commit()
def add_staff(user_obj):
items = session.query(User).filter(User.name == user_obj.name).all()
if len(items) == 0: # 数据库中没有重名的存在，则添加
session.add(user_obj)
session.commit()
else:
# 弹出警告
ToolsManager.information_box(u"注意", u"\"%s\"已经存在数据库中!" % unicode(user_obj.name))
def updata_staff(dic):
item = session.query(User).filter(User.id == dic.get('id')).one()
item.name = dic.get('name')
item.employee_id = dic.get('employee_id')
item.phone_number = dic.get('phone_number')
item.tel_number = dic.get('tel_number')
item.birth_date = dic.get('birth_date')
item.title = dic.get('title')
item.position = dic.get('position')
item.education = dic.get('education')
item.description = dic.get('description')
session.add(item)
session.commit()
def updata_project(dic):
item = session.query(Project).filter(Project.id == dic.get('id')).one()
item.name = dic.get('name')
item.search_id = dic.get('search_id')
item.source_place = dic.get('source_place')
item.main_designer = dic.get('main_designer')
item.design_all = dic.get('design_all')
item.responsible_man = dic.get('responsible_man')
item.start_time = dic.get('start_time')
item.end_time = dic.get('end_time')
item.description = dic.get('description')
session.add(item)
session.commit()
StaffManager.delete_staff_project_by_project_ids([item.id])
StaffManager.add_staff_project(item.id, dic.get('attendee_ids'))
def delete_project(ids_list):
session.query(Project).filter(Project.id.in_(ids_list)).delete(synchronize_session=False)
session.commit()
def delete_staff(ids_list):
session.query(User).filter(User.id.in_(ids_list)).delete(synchronize_session=False)
session.commit()
def delete_staff_project_by_staff_ids(user_ids):
session.query(UserProject).filter(UserProject.user_id.in_(user_ids)).delete(synchronize_session=False)
session.commit()
