def get_stage(message): database.create_users_db() # open connection conn = sqlite3.connect(config.db_name) cursor = conn.cursor() a = cursor.execute( f"SELECT reg_stage FROM {config.users_table} WHERE id='{message.chat.id}'" ).fetchall() if len(a) > 0: a = a[0][0] # close connection conn.commit() cursor.close() conn.close() return a # close connection conn.commit() cursor.close() conn.close()
def create_user(message): # checking for existence database.create_users_db() # open connection conn = sqlite3.connect(config.db_name) cursor = conn.cursor() # Adding a new user if does not exist a = len( cursor.execute( f"SELECT * FROM {config.users_table} WHERE id='{message.chat.id}'" ).fetchall()) if a == 0: cursor.execute( f"INSERT INTO {config.users_table} (id, rating, reg_stage) VALUES ('{message.chat.id}', {0}, {1})" ) bot.send_message(message.chat.id, messages.USERREG) bot.send_message(message.chat.id, messages.USERNAME) else: bot.send_message(message.chat.id, messages.HELP) # close connection conn.commit() cursor.close() conn.close()
import config import voice import database import messages import telebot import sqlite3 from telebot import types import random import math import numpy bot = telebot.TeleBot(config.token) database.create_users_db() database.create_estabs_db() def get_stage(message): database.create_users_db() # open connection conn = sqlite3.connect(config.db_name) cursor = conn.cursor() a = cursor.execute( f"SELECT reg_stage FROM {config.users_table} WHERE id='{message.chat.id}'" ).fetchall() if len(a) > 0: a = a[0][0] # close connection conn.commit() cursor.close()
from flask import Flask, render_template, request, session, flash, redirect, url_for from flask_session import Session from sqlalchemy import create_engine from sqlalchemy.orm import scoped_session, sessionmaker import os import random #local imports from database import create_users_db, register_user, username_exist, login_user #create users database if not exists create_users_db() # making a flask app app = Flask(__name__) # configure app before passing it to session app.config['SESSION_TYPE'] = 'filesystem' app.config["SESSION_PERMANENT"] = False Session(app) # setting up SQLAlcehmy engine # an engine is an object created by SQLAlchemy to take care of interacting with the database engine = create_engine(os.getenv("DATABASE_URL")) # creates different sessions for different users db = scoped_session(sessionmaker(bind=engine)) app.secret_key = os.urandom(24) #number of quotes on database quotes_number = db.execute("SELECT COUNT(*) FROM quotes").fetchone()[0] @app.route("/")