def google_update(message: Message):
if not is_person(message.chat):
return
try:
user = User.get(telegram_id=message.chat.id)
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Admins'):
raise Exception("Unauthorized user")
except Exception:
bot.send_message(text=get_unauthorized_user_start_message(),
chat_id=message.chat.id)
return
cur_equipments = Equipment.select()
cur_movement = Movement.select()
gs = GoogleSync()
equipments_from_google = gs.read_range(
list_name='Список оборудования',
range_in_list=f'A{cur_equipments.count() + 2}:G')
movements_from_google = gs.read_range(
list_name='Перемещение оборудования',
range_in_list=f'A{cur_movement.count() + 2}:C')
if equipments_from_google is not None:
if len(equipments_from_google) > 0:
for item in equipments_from_google:
if len(item) < 7:
for j in range(len(item), 7):
item.append('')
Equipment.create(it_id=item[0],
pos_in_buh=item[1],
invent_num=item[2],
type=item[3],
mark=item[4],
model=item[5],
serial_num=item[6])
if movements_from_google is not None:
if len(movements_from_google) > 0:
for item in movements_from_google:
if len(item) < 7:
for j in range(len(item), 7):
item.append('')
if item[0] == '':
continue
Movement.create(equipment=Equipment.get(it_id=item[0]),
campus=item[1],
room=item[2])
# cur_persons_count = Person.select().count()
# gs_phones = GoogleSync(spreadsheet_id=PHONE_SPREADSHEET_ID)
# persons_from_google = gs_phones.read_range(list_name='List1',
# range_in_list=f'A{cur_persons_count + 2}:F')
# for person in persons_from_google:
# if len(person) < 6:
# for j in range(len(person), 6):
# person.append('')
# Person.get_or_create(name=person[1],
# surname=person[0],
# patronymic=person[2],
# defaults={
# 'position': person[3],
# 'phone': f'+{person[4]}',
# 'email': person[5],
# 'photo': '',
# 'actual': 'True'
# })
bot.send_message(chat_id=user.telegram_id, text='Данные получены')
def plain_text(message: Message):
if not is_person(message.chat):
return
try:
user = User.get(telegram_id=message.chat.id)
if user in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Unauthorized'):
raise Exception("Unauthorized user")
except Exception:
bot.send_message(text=get_unauthorized_user_start_message(),
chat_id=message.chat.id)
return
if user.status == 'Adding group':
if user in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Admins'):
group, created = Group.get_or_create(group_name=message.text)
bot.send_message(chat_id=message.chat.id, text='Группа добавлена')
groups = Group.select()
return_str = 'Список групп:\n'
for group in groups:
return_str += group.group_name + '\n'
bot.send_message(text=return_str, chat_id=message.chat.id)
logger.info(
f'Admin {user.first_name} {user.last_name} add new group - {message.text}'
)
elif user.status == 'zavhoz_check_equipment':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Zavhoz'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
looking_invent_num = message.text
try:
Equipment.get(invent_num=looking_invent_num)
except Exception as e:
logger.info(
f'User {user.first_name} {user.last_name} looking for unexisting equipment with {looking_invent_num} invent number'
)
bot.send_message(
chat_id=user.telegram_id,
text=
f'Оборудование с инвентарным номером {looking_invent_num} не стоит на балансе'
)
User.update(status='').where(User.id == user.id).execute()
return
found_equipments = Equipment.select().where(
Equipment.invent_num == looking_invent_num)
for item in found_equipments:
bot.send_message(chat_id=user.telegram_id,
text=equipment_info(equipment=item))
logger.info(
f'User {user.first_name} {user.last_name} looked info about {looking_invent_num} invent number'
)
elif user.status == 'invent_search':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
invent_num = message.text
found_equipments = Equipment.select().where(
Equipment.invent_num == invent_num)
if found_equipments.count() == 0:
bot.send_message(
chat_id=user.telegram_id,
text=
f'Оборудование с инвентарным номером {invent_num} не стоит на балансе'
)
else:
for equipment in found_equipments:
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
elif user.status == 'serial_search':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
serial_num = message.text
found_equipments = Equipment.select().where(
Equipment.serial_num == serial_num)
if found_equipments.count() == 0:
bot.send_message(
chat_id=user.telegram_id,
text=
f'Оборудование с серийным номером {serial_num} не стоит на балансе'
)
else:
for equipment in found_equipments:
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
elif user.status.split('_')[0] == 'edit-type':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
equipment = Equipment.get(id=user.status.split('_')[1])
Equipment.update(type=message.text).where(
Equipment.id == equipment.id).execute()
equipment = Equipment.get(id=user.status.split('_')[1])
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
send_equipment_info_to_google_sheet(equipment)
logger.info(
f'User {user.first_name} {user.last_name} edit equipment ID {equipment.it_id}: new type is {equipment.type}'
)
elif user.status.split('_')[0] == 'edit-mark':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
equipment = Equipment.get(id=user.status.split('_')[1])
Equipment.update(mark=message.text).where(
Equipment.id == equipment.id).execute()
equipment = Equipment.get(id=user.status.split('_')[1])
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
send_equipment_info_to_google_sheet(equipment)
logger.info(
f'User {user.first_name} {user.last_name} edit equipment ID {equipment.it_id}: new mark is {equipment.mark}'
)
elif user.status.split('_')[0] == 'edit-model':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
equipment = Equipment.get(id=user.status.split('_')[1])
Equipment.update(model=message.text).where(
Equipment.id == equipment.id).execute()
equipment = Equipment.get(id=user.status.split('_')[1])
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
send_equipment_info_to_google_sheet(equipment)
logger.info(
f'User {user.first_name} {user.last_name} edit equipment ID {equipment.it_id}: new model is {equipment.model}'
)
elif user.status.split('_')[0] == 'edit-serial':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
equipment = Equipment.get(id=user.status.split('_')[1])
Equipment.update(serial_num=message.text).where(
Equipment.id == equipment.id).execute()
equipment = Equipment.get(id=user.status.split('_')[1])
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
send_equipment_info_to_google_sheet(equipment)
logger.info(
f'User {user.first_name} {user.last_name} edit equipment ID {equipment.it_id}: new serial num is {equipment.serial_num}'
)
elif user.status.split('/')[0] == 'create_movement':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Inventarization'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
campus = user.status.split('/')[1].split('-')[1]
room = message.text
equipment = Equipment.get(
id=user.status.split('/')[2].split('-')[1])
movement = Movement.create(
equipment=equipment,
campus=f'УК {campus}'
if not campus == 'spisanie' else 'Списание',
room=room)
equipment = Equipment.get(
id=user.status.split('/')[2].split('-')[1])
bot.send_message(
chat_id=user.telegram_id,
text=equipment_info(equipment),
reply_markup=get_equipment_reply_markup(equipment))
send_movement_to_google_sheet(equipment, movement)
logger.info(
f'User {user.first_name} {user.last_name} move equipment ID {equipment.it_id}: new location is {movement.campus} {movement.room}'
)
elif user.status.split(':')[0] == 'Edit_person_info':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'PhonesAdmin'):
bot.send_message(chat_id=user.telegram_id,
text='У Вас нет доступа к этой функции')
logger.info(
f'User {user.first_name} {user.last_name} had unsupported status!'
)
else:
edit_parameter = user.status.split(':')[1].split('_')[0]
person = Person.get(id=user.status.split(':')[1].split('_')[1])
if edit_parameter == 'surname':
Person.update(surname=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'name':
Person.update(name=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'patronymic':
Person.update(patronymic=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'phone':
Person.update(phone=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'position':
Person.update(position=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'email':
Person.update(email=message.text).where(
Person.id == person.id).execute()
elif edit_parameter == 'photo':
bot.send_message(
chat_id=message.chat.id,
text=
'Для редактирования фото сотрудника необходимо прислать фото, а не текст!'
)
return
person = Person.get(id=user.status.split(':')[1].split('_')[1])
update_person_info_in_google(person)
if not person.photo == '':
bot.send_photo(chat_id=message.chat.id,
photo=person.photo,
caption=get_person_info(person),
reply_markup=get_contact_reply_markup(
user, person))
else:
bot.send_message(chat_id=message.chat.id,
text=get_person_info(person),
reply_markup=get_contact_reply_markup(
user, person))
bot.send_contact(chat_id=message.chat.id,
phone_number=person.phone,
first_name=person.surname,
last_name=f"{person.name} {person.patronymic}")
elif user.status.split('/')[0] == 'phone_search':
search_parameter = user.status.split('/')[1]
template = message.text
founded_persons = None
if search_parameter == 'surname':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'PhonesAdmin'):
founded_persons = Person.select().where(
Person.surname == template).where(Person.actual == 'True')
else:
founded_persons = Person.select().where(
Person.surname == template)
elif search_parameter == 'name':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'PhonesAdmin'):
founded_persons = Person.select().where(
Person.name == template.split(' ')[0]
and Person.patronymic == template.split(' ')[1]).where(
Person.actual == 'True')
else:
founded_persons = Person.select().where(
Person.name == template.split(' ')[0]
and Person.patronymic == template.split(' ')[1])
elif search_parameter == 'number':
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'PhonesAdmin'):
founded_persons = Person.select().where(
Person.phone == template).where(Person.actual == 'True')
else:
founded_persons = Person.select().where(
Person.phone == template)
if user not in User.select(User).join(Links).join(Group).where(
Group.group_name == 'PhonesAdmin'):
founded_persons = founded_persons.where(Person.actual == 'True')
if founded_persons.count() > 0:
for person in founded_persons:
if not person.photo == '':
bot.send_photo(chat_id=message.chat.id,
photo=person.photo,
caption=get_person_info(person),
reply_markup=get_contact_reply_markup(
user, person))
else:
bot.send_message(chat_id=message.chat.id,
text=get_person_info(person),
reply_markup=get_contact_reply_markup(
user, person))
bot.send_contact(
chat_id=message.chat.id,
phone_number=person.phone,
first_name=person.surname,
last_name=f"{person.name} {person.patronymic}")
else:
bot.send_message(chat_id=message.chat.id,
text='Я никого не нашел по введенным Вами данным')
else:
bot.send_message(
chat_id=message.chat.id,
text=
'Воспользуйтесь кнопками или командами (/help) для выбора функции')
User.update(status='').where(User.id == user.id).execute()
def equipment_exists(form, field):
if Equipment.select().where(Equipment.unitnumber == field.data).exists():
raise ValidationError('Equipment with that unit number already exists.')