Advertisement
Platitude

Untitled

Mar 14th, 2020
194
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 4.20 KB | None | 0 0
  1. import requests, telebot, logging
  2. from telebot import types, apihelper
  3. from telegram import Update
  4. from telegram.ext import Updater, CommandHandler
  5. from telegram import ReplyKeyboardMarkup
  6.  
  7. import mysql.connector
  8.  
  9. mydb = mysql.connector.connect(
  10. host="localhost",
  11. user="root",
  12. passwd="0000",
  13. port="3306",
  14. database="*******"
  15. )
  16. mycursor = mydb.cursor()
  17. bot = telebot.TeleBot('Token')
  18. user_data = {}
  19.  
  20. @bot.message_handler(commands=['help'])
  21. def help_list(message):
  22. a = ['/start', '/reg_employer', '/help', '/url', "/switch"]
  23. for i in a:
  24. bot.send_message(message.chat.id, i)
  25.  
  26.  
  27. class User:
  28. def __init__(self, first_name):
  29. self.first_name = first_name
  30. self.last_name = None
  31. self.middle_name = None
  32. self.phone_number = None
  33. self.subjects_civil_law = None
  34. self.region = None
  35. self.services_provided = None
  36.  
  37.  
  38. @bot.message_handler(commands=['reg_employer']) # получение команды
  39. def send_welcome(message):
  40. msg = bot.send_message(message.chat.id, "Введите Ваше имя: ")
  41. bot.register_next_step_handler(msg, process_lastname_step)
  42.  
  43.  
  44. def process_lastname_step(message):
  45. user_id = message.from_user.id
  46. user_data[user_id] = User(message.text)
  47. msg = bot.send_message(message.chat.id, "Введите Вашу фамилию: ")
  48. bot.register_next_step_handler(msg, process_middlename_step)
  49.  
  50.  
  51. def process_middlename_step(message):
  52. user_id = message.from_user.id
  53. user_data[user_id] = User(message.text)
  54. msg = bot.send_message(message.chat.id, "Введите Вашу отчество: ")
  55. bot.register_next_step_handler(msg, process_phonenumber_step)
  56.  
  57.  
  58. def process_phonenumber_step(message):
  59. user_id = message.from_user.id
  60. user_data[user_id] = User(message.text)
  61. msg = bot.send_message(message.chat.id, "Введите Вашу номер телефона: ")
  62. bot.register_next_step_handler(msg, process_subjects_civil_law_step)
  63.  
  64.  
  65. def process_subjects_civil_law_step(message):
  66. user_id = message.from_user.id
  67. user_data[user_id] = User(message.text)
  68. msg = bot.send_message(message.chat.id, "Укажите вы являетесь юридическим или физическим лицом: ")
  69. bot.register_next_step_handler(msg, process_region_step)
  70.  
  71.  
  72. def process_region_step(message):
  73. user_id = message.from_user.id
  74. user_data[user_id] = User(message.text)
  75. msg = bot.send_message(message.chat.id, "Укажите Ваш город: ")
  76. bot.register_next_step_handler(msg, process_services_provided_step)
  77.  
  78.  
  79. def process_services_provided_step(message):
  80. user_id = message.from_user.id
  81. user_data[user_id] = User(message.text)
  82. msg = bot.send_message(message.chat.id,
  83. "Каким выдом деятельности вы занимаетесь (стройка, программирования, разработка систем, оказание услуг): ")
  84. bot.register_next_step_handler(msg, process_lastname_step)
  85.  
  86.  
  87. def process_lastname_step(message):
  88. #try:
  89. user_id = message.from_user.id
  90. user = user_data[user_id]
  91.  
  92. sql = "INSERT INTO users_employer (first_name, last_name, middle_name, phone_number, subjects_civil_law, region, services_provided, user_id) " \
  93. "VALUES (%s, %s, %s, %s, %s, %s, %s, %s)"
  94. val = (
  95. user.first_name, user.last_name, user.middle_name, user.phone_number, user.subjects_civil_law, user.region,
  96. user.services_provided, user.user_id)
  97. mycursor.execute(sql, val)
  98. mydb.commit()
  99. bot.send_message(message.chat.id, "Вы зарегестрированны")
  100. #except Exception as e: \
  101. #bot.send_message(message.chat.id, "Вы уже зарегестрированны")
  102.  
  103.  
  104. @bot.message_handler(content_types=['text'])
  105. def get_text_messages(message):
  106. if message.text == "Привет":
  107. bot.send_message(message.from_user.id, "Привет, чем я могу тебе помочь?")
  108. else:
  109. bot.send_message(message.from_user.id, "Я тебя не понимаю.")
  110.  
  111.  
  112. bot.polling() # запуск бота
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement