# SQLite comes built-in with Python! import sqlite3 # Connect to database (creates if doesn't exist) conn = sqlite3.connect('joke_bot.db') cursor = conn.cursor() # Create a table cursor.execute('''CREATE TABLE jokes ( id INTEGER PRIMARY KEY, text TEXT NOT NULL, added_by TEXT, likes INTEGER DEFAULT 0, dislikes INTEGER DEFAULT 0, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP )''') conn.commit() conn.close()

# Create a new file: database.py # This will handle all database operations import sqlite3 from datetime import datetime class JokeDatabase: def __init__(self, db_name='joke_bot.db'): self.db_name = db_name self.init_database() def get_connection(self): """Get a database connection""" conn = sqlite3.connect(self.db_name) conn.row_factory = sqlite3.Row # Access columns by name return conn def init_database(self): """Initialize database tables""" conn = self.get_connection() cursor = conn.cursor() # Create jokes table cursor.execute('''CREATE TABLE IF NOT EXISTS jokes ( id INTEGER PRIMARY KEY AUTOINCREMENT, joke_text TEXT NOT NULL, user_id INTEGER, username TEXT, likes INTEGER DEFAULT 0, dislikes INTEGER DEFAULT 0, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP )''') # Create ratings table to track user votes cursor.execute('''CREATE TABLE IF NOT EXISTS ratings ( user_id INTEGER, joke_id INTEGER, vote INTEGER, -- 1 for like, -1 for dislike PRIMARY KEY (user_id, joke_id), FOREIGN KEY (joke_id) REFERENCES jokes(id) )''') conn.commit() conn.close() # Add some default jokes if table is empty self.add_default_jokes() def add_default_jokes(self): """Add default jokes if database is empty""" conn = self.get_connection() cursor = conn.cursor() cursor.execute("SELECT COUNT(*) FROM jokes") count = cursor.fetchone()[0] if count == 0: default_jokes = [ ("Why did the robot go to school? To recharge his brain! 🔋", 0, "System"), ("Knock knock!\nWho's there?\nLettuce!\nLettuce who?\nLettuce in!", 0, "System"), ("Why don't eggs tell jokes? They'd crack each other up! 🥚", 0, "System") ] cursor.executemany('''INSERT INTO jokes (joke_text, user_id, username) VALUES (?, ?, ?)''', default_jokes) conn.commit() print(f"Added {len(default_jokes)} default jokes to database") conn.close()

# Add these methods to the JokeDatabase class def add_joke(self, joke_text, user_id, username): """Add a new joke to the database""" conn = self.get_connection() cursor = conn.cursor() cursor.execute('''INSERT INTO jokes (joke_text, user_id, username) VALUES (?, ?, ?)''', (joke_text, user_id, username)) joke_id = cursor.lastrowid conn.commit() conn.close() return joke_id def get_random_joke(self): """Get a random joke, weighted by rating""" conn = self.get_connection() cursor = conn.cursor() # Weighted random: jokes with higher likes appear more often cursor.execute('''SELECT * FROM jokes ORDER BY (likes - dislikes + 5) * RANDOM() DESC LIMIT 1''') joke = cursor.fetchone() conn.close() return dict(joke) if joke else None def get_joke_by_id(self, joke_id): """Get a specific joke by ID""" conn = self.get_connection() cursor = conn.cursor() cursor.execute("SELECT * FROM jokes WHERE id = ?", (joke_id,)) joke = cursor.fetchone() conn.close() return dict(joke) if joke else None def get_top_jokes(self, limit=5): """Get top-rated jokes""" conn = self.get_connection() cursor = conn.cursor() cursor.execute('''SELECT * FROM jokes ORDER BY (likes - dislikes) DESC, likes DESC LIMIT ?''', (limit,)) jokes = [dict(row) for row in cursor.fetchall()] conn.close() return jokes def get_total_jokes(self): """Get total number of jokes""" conn = self.get_connection() cursor = conn.cursor() cursor.execute("SELECT COUNT(*) FROM jokes") count = cursor.fetchone()[0] conn.close() return count

# Add rating methods to JokeDatabase class def rate_joke(self, user_id, joke_id, vote): """ Rate a joke (like or dislike) vote: 1 for like, -1 for dislike, 0 to remove rating """ conn = self.get_connection() cursor = conn.cursor() # Check if user already voted cursor.execute('''SELECT vote FROM ratings WHERE user_id = ? AND joke_id = ?''', (user_id, joke_id)) existing_vote = cursor.fetchone() if existing_vote: old_vote = existing_vote[0] if vote == 0: # Remove vote cursor.execute('''DELETE FROM ratings WHERE user_id = ? AND joke_id = ?''', (user_id, joke_id)) # Update joke counts if old_vote == 1: cursor.execute('''UPDATE jokes SET likes = likes - 1 WHERE id = ?''', (joke_id,)) elif old_vote == -1: cursor.execute('''UPDATE jokes SET dislikes = dislikes - 1 WHERE id = ?''', (joke_id,)) elif old_vote != vote: # Change vote cursor.execute('''UPDATE ratings SET vote = ? WHERE user_id = ? AND joke_id = ?''', (vote, user_id, joke_id)) # Update joke counts (remove old, add new) if old_vote == 1 and vote == -1: cursor.execute('''UPDATE jokes SET likes = likes - 1, dislikes = dislikes + 1 WHERE id = ?''', (joke_id,)) elif old_vote == -1 and vote == 1: cursor.execute('''UPDATE jokes SET likes = likes + 1, dislikes = dislikes - 1 WHERE id = ?''', (joke_id,)) else: if vote != 0: # New vote cursor.execute('''INSERT INTO ratings (user_id, joke_id, vote) VALUES (?, ?, ?)''', (user_id, joke_id, vote)) # Update joke counts if vote == 1: cursor.execute('''UPDATE jokes SET likes = likes + 1 WHERE id = ?''', (joke_id,)) elif vote == -1: cursor.execute('''UPDATE jokes SET dislikes = dislikes + 1 WHERE id = ?''', (joke_id,)) conn.commit() # Get updated joke info cursor.execute("SELECT likes, dislikes FROM jokes WHERE id = ?", (joke_id,)) result = cursor.fetchone() conn.close() return { 'likes': result[0] if result else 0, 'dislikes': result[1] else 0 } def get_user_vote(self, user_id, joke_id): """Get user's vote for a specific joke""" conn = self.get_connection() cursor = conn.cursor() cursor.execute('''SELECT vote FROM ratings WHERE user_id = ? AND joke_id = ?''', (user_id, joke_id)) result = cursor.fetchone() conn.close() return result[0] if result else 0

# Updated app.py with database integration from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters, CallbackQueryHandler from database import JokeDatabase import random # Initialize database db = JokeDatabase() async def start(update: Update, context: ContextTypes.DEFAULT_TYPE): """Start command handler""" welcome_text = """🤖 Welcome to Joke Bot 2.0! 🤖 Commands: /joke - Get a random joke /addjoke - Submit your own joke /top - See top-rated jokes /like - Like the last joke /dislike - Dislike the last joke /stats - Bot statistics Now with database, user submissions, and ratings!""" await update.message.reply_text(welcome_text) async def send_joke(update: Update, context: ContextTypes.DEFAULT_TYPE): """Send a random joke from database""" joke = db.get_random_joke() if not joke: await update.message.reply_text("No jokes in database yet! Use /addjoke to add one.") return # Store joke ID in context for rating context.user_data['last_joke_id'] = joke['id'] # Format joke with rating info rating_text = f"👍 {joke['likes']} 👎 {joke['dislikes']}" joke_text = f"{joke['joke_text']}\n\n{rating_text}" # Add inline buttons for quick rating keyboard = [ [ InlineKeyboardButton("👍 Like", callback_data=f"like_{joke['id']}"), InlineKeyboardButton("👎 Dislike", callback_data=f"dislike_{joke['id']}") ] ] reply_markup = InlineKeyboardMarkup(keyboard) await update.message.reply_text(joke_text, reply_markup=reply_markup) async def add_joke_command(update: Update, context: ContextTypes.DEFAULT_TYPE): """Handle /addjoke command""" if not context.args: await update.message.reply_text( "Please provide a joke after the command:\n" "Example: /addjoke Why did the chicken cross the road?" ) return joke_text = " ".join(context.args) user = update.effective_user # Add to database joke_id = db.add_joke(joke_text, user.id, user.username) await update.message.reply_text( f"✅ Joke added successfully! (ID: {joke_id})\n" "Others can now rate it with /like and /dislike" )