def query_text(message):
roles = {
'Житель города': {
'greet': 'Отлично, Вы житель!',
'role': 0
},
'Старший по дому': {
'greet': 'Добро пожаловать!',
'role': 1
},
'Сотрудник УК/ТСЖ': {
'greet': 'Приятно поработать!',
'role': 2
},
}
with open(settings.MEDIA_ROOT + f'{message.chat.id}.txt', 'a') as f:
f.writelines([str(roles[message.text]['role']) + '\n'])
bot.send_message(message.chat.id, roles[message.text]['greet'])
if message.text != 'Сотрудник УК/ТСЖ':
markup = types.ReplyKeyboardMarkup(row_width=1)
geo_button = types.KeyboardButton(text="Отправить местоположение",
request_location=True)
markup.add(geo_button)
bot.send_message(message.chat.id,
"Нажмите на кнопку и передайте координаты.",
reply_markup=markup)
bot.register_next_step_handler(message, location)
else:
lines = [
line
for line in open(settings.MEDIA_ROOT + f'{message.chat.id}.txt')
]
employee = Employee(
external_id=lines[0],
surname=lines[1],
name=lines[2],
tg_role=roles[message.text]['role'],
position_id=random.choice([1, 2])
) # TODO Хреновая реализация, но для ускорения разработки пока так
employee.save()
org = Organization.objects.get(pk=1)
employee.organizations.add(org)
delete_file(settings.MEDIA_ROOT + f'{message.chat.id}.txt')
bot.send_message(message.chat.id,
"Здесь Вы будете получать заявки для выполнения.",
reply_markup=types.ReplyKeyboardRemove())
pAccount3.save()
pAccount4 = Contact(name="Jane",
title="staff",
address="Tainan",
phone="063234567",
mobile="0987654321",
ssn="E123456744",
birthday="1980-04-01")
pAccount4.save()
pAccountC1 = Company(name="Company A", )
pAccountC1.save()
pAccountC2 = Company(name="Company B", )
pAccountC2.save()
pAccountE1 = Employee(contact=pAccount1)
pAccountE1.save()
pAccountE2 = Employee(contact=pAccount2)
pAccountE2.save()
pAccountE3 = Employee(contact=pAccount3)
pAccountE3.save()
pAccountE4 = Employee(contact=pAccount4)
pAccountE4.save()
#project設定
pProject1 = Project(contact=pAccount1, company=pAccountC1, name="SkyTower")
pProject1.save()
pProject2 = Project(contact=pAccount2, company=pAccountC2, name="MoonLand")
pProject2.save()
pProjectA1 = Assignment(project=pProject1,