def get_start(message: Message):
if not is_person(message.chat):
return
try:
user = User.get(telegram_id=str(message.chat.id))
if user in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Unauthorized'):
raise Exception("Unauthorized user")
except Exception:
logger.info('Unauthorized user')
user, created = User.get_or_create(
telegram_id=message.chat.id,
first_name=message.from_user.first_name
if message.from_user.first_name is not None else '',
last_name=message.from_user.last_name
if message.from_user.last_name is not None else '',
status='waiting for access')
unauth_group, created = Group.get_or_create(group_name='Unauthorized')
Links.create(user=user, group=unauth_group)
bot.send_message(text=get_unauthorized_user_start_message(),
chat_id=message.chat.id)
for admin in User.select(User).join(Links).join(Group).where(
Group.group_name == 'Admins'):
bot.send_message(text=get_new_unauthorized_user_message(user),
chat_id=admin.telegram_id,
reply_markup=keyboard_to_chose_users_groups(user))
return
bot.send_message(
text=f'С возвращением, {user.first_name} {user.last_name}',
chat_id=message.chat.id,
reply_markup=get_start_keyboard(user))
def create():
context = get_context()
context.update(dict(success=False))
user = User.get_or_create(email=context['email'])
all_users = User.get_all(exclude_user_id=user.id)
context.update({'all_users': list(all_users)})
if request.method == 'POST':
group = Group.get_or_create(name=request.form['group_name'], status='Pending')
UserGroup.get_or_create(
user_id=user.id,
group_id=group.id,
loan_amount=request.form['loan_amount'],
repaid_amount='0',
)
all_users = User.get_all(exclude_user_id=user.id)
for _user in all_users:
UserGroup.get_or_create(
user_id=_user.id,
group_id=group.id,
loan_amount='0',
repaid_amount='0',
)
context.update(dict(success=True))
return render_template('create.html', **context)
def getConversation(self, jid):
if self.isGroupJid(jid):
group = Group.get_or_create(jid = jid)
conversation = Conversation.get_or_create(group = group)[0]
else:
contact = Contact.get_or_create(jid = jid)
conversation = Conversation.get_or_create(contact = contact[0])[0]
return conversation
def seed_event(obj):
if (not have_keywords(obj["name"])) and (not have_keywords(
obj["description"])):
return
# create group
url = "https://www.meetup.com/" + obj["group"]["urlname"]
try:
photo_url = obj["group"]["group_photo"]["thumb_link"]
except KeyError:
photo_url = ""
group, group_created = Group.get_or_create(id=str(uuid.uuid4()),
source="MEETUP",
name=obj["group"]["name"],
url=url,
photoUrl=photo_url,
sourceId=obj["group"]["id"])
if group_created:
print("A new group '{}' is created successfully!".format(group.name))
# create event
try:
photo_url = obj["photo_url"]
except KeyError:
photo_url = ""
time_local = datetime.utcfromtimestamp(
int(obj["time"]) / 1000 +
int(obj["utc_offset"]) / 1000).strftime("%m/%d/%Y, %H:%M:%S")
try:
country = obj["venue"]["country"]
except KeyError:
country = ""
try:
city = obj["venue"]["city"]
except KeyError:
city = ""
try:
lat = obj["venue"]["lat"]
except KeyError:
lat = ""
try:
lon = obj["venue"]["lon"]
except KeyError:
lon = ""
try:
event, event_created = Event.get_or_create(
id=str(uuid.uuid4()),
source="MEETUP",
name=obj["name"],
description=obj["description"],
url=obj["event_url"],
photoUrl=photo_url,
sourceId=obj["id"],
time=datetime.fromtimestamp(int(obj["time"]) / 1000),
timeLocal=time_local,
country=country,
city=city,
lat=lat,
lon=lon,
group_id=group.id)
if event_created:
print("A new event '{}' is created successfully! ".format(
event.name))
except Exception as err:
print(err)
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()
