EduSearch/examples/tm_bot.wsgi

# -*- encoding: utf-8 -*-
# 
import os, json, sys, subprocess, urllib.parse, traceback

virtual_env = os.path.expanduser('~/projects/world-it-planet/env')
activate_this = os.path.join(virtual_env, 'bin/activate_this.py')
exec(open(activate_this).read(), dict(__file__=activate_this))


import random, time, datetime, os, re
import telegram

#декларативное определение
from sqlalchemy import Column, Integer, String, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

#---------------------------------- Variables ----------
areas = {"bryansk":19}
specializations = ['программист', 'стажер', 'стажировка']

#максимальное количество публикуемых вакансий за 1 раз
max_vacancies = 3

access_token = "1775925477:AAH1Wlk22hxjSghqOH_IEuolj_FWO2k_YUs"
chat_id = "-1001166215020"
#---------------------------------- Variables End ----------




def application(env, start_response):
    out_s = ""
    #Инициализация SQLLite
    basedir = os.path.abspath(os.path.dirname(__file__))
    SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(basedir, 'hhtm.db')
    engine = create_engine(SQLALCHEMY_DATABASE_URI, pool_pre_ping=True)

    Base = declarative_base()
    class Vacancies(Base):
        __tablename__ = 'vacancies'
        id = Column(Integer, primary_key=True, autoincrement=True)
        title = Column(String(512))
        city = Column(String(20))
        specialization = Column(String(255))
        href = Column(String(512))
        donor = Column(String(255))
        vacancy_id = Column(Integer)
        vacancy_date = Column(Integer)
        parse_date = Column(Integer)
        employer = Column(String(255))
        canal_city_id = Column(Integer)
        canal_city_date = Column(Integer)
        canal_spec_id = Column(Integer)
        canal_spec_date = Column(Integer)

        def __init__(self, title, city, specialization, href, donor, vacancy_id, vacancy_date, parse_date, employer, canal_city_id, canal_city_date, canal_spec_id, canal_spec_date):
            self.title = title
            self.city = city
            self.specialization = specialization
            self.href = href
            self.donor = donor
            self.vacancy_id = vacancy_id
            self.vacancy_date = vacancy_date
            self.parse_date = parse_date
            self.employer = employer
            self.canal_city_id = canal_city_id
            self.canal_city_date = canal_city_date
            self.canal_spec_id = canal_spec_id
            self.canal_spec_date = canal_spec_date

        def __repr__(self):
            return "<Vacancy('%s','%s', '%s')>" % (self.title, self.specialization, self.href)


    class Log(Base):
        __tablename__ = 'log'
        id = Column(Integer, primary_key=True, autoincrement=True)
        action = Column(String(64))
        status = Column(String(64))
        time = Column(Integer)
        donor = Column(String(64))
        city = Column(String(20))
        specialization = Column(String(20))
        vacancies_count = Column(Integer)
        canal_id = Column(String(64))

        def __init__(self, action, status, time, donor, city, specialization, vacancies_count, canal_id):
            self.action = action
            self.status = status
            self.time = time
            self.donor = donor
            self.city = city
            self.specialization = specialization
            self.vacancies_count = vacancies_count
            self.canal_id = canal_id

        def __repr__(self):
            return "<Log('%s','%s', '%s')>" % (self.action, self.status)


    Session = sessionmaker(bind=engine)
    session = Session()

    city, area = random.choice(list(areas.items())) 
    specialization = random.choice(specializations)

    now = datetime.datetime.now()
    hour_now = int(now.strftime("%H"))

    if ((hour_now > 8) and (hour_now < 20)):
        vacancy = session.query(Vacancies).filter(Vacancies.canal_city_date == 0).first()
        try:
            bot = telegram.Bot(token=access_token)
            text = vacancy.title + ": <a href='" + vacancy.href + "'>" + vacancy.href + "</a>"
            tm_response = str(bot.sendMessage(chat_id=chat_id, text=text, parse_mode=telegram.ParseMode.HTML))
            #ответ не стандартный json
            tm_response = tm_response.replace("\\", "\\\\")
            tm_response = tm_response.strip("'<>() ").replace("\'", "\"")
            tm_response = tm_response.replace("False", "\"False\"")
            message_id = int(json.loads(tm_response)["message_id"])
            #удаляем следующее сообщение
            time.sleep(3)
            message_id = message_id + 1
            bot.delete_message(chat_id=chat_id, message_id=message_id)
            status = "Ok test"
            vacancy.canal_city_date = int(time.time())
        except:
            status = str(traceback.format_exc())
            out_s = str(status)
            out_s = out_s + "<br>" + tm_response




        new_log = Log(
            action = "post", 
            status = status, 
            time = int(time.time()),
            donor = 'hh.ru', 
            city = city, 
            specialization = specialization,
            vacancies_count = 0,
            canal_id = chat_id, 
            )
        session.add(new_log)



    session.commit()
    start_response('200 OK', [('Content-Type','text/html')])
    b = out_s.encode('utf-8')
    return [b]