import uuid import sqlite3 # Connect to database conn = sqlite3.connect('users.db') c = conn.cursor() # User registration system def register_user(name, email, password): # Generate unique user ID user_id = uuid.uuid4().hex # Insert user into database c.execute("INSERT INTO users (id, name, email, password) VALUES (?, ?, ?, ?)", (user_id, name, email, password)) conn.commit() # Generate referral link referral_link = "https://example.com/signup?ref={}".format(user_id) return referral_link # Track referrals def track_referral(referral_id): # Check if referral already exists c.execute("SELECT * FROM referrals WHERE id = ?", (referral_id,)) referral = c.fetchone() if referral: # Referral already exists return False else: # Add referral to database c.execute("INSERT INTO referrals (id, user_id) VALUES (?, ?)", (referral_id, referral_id)) conn.commit() # Credit referrer's account with reward c.execute("UPDATE users SET balance = balance + 10 WHERE id = ?", (referral_id,)) conn.commit() return True # Display referral stats def get_referral_stats(user_id): # Get number of referrals c.execute("SELECT COUNT(*) FROM referrals WHERE user_id = ?", (user_id,)) num_referrals = c.fetchone()[0] # Get total rewards earned c.execute("SELECT SUM(reward) FROM referrals WHERE user_id = ?", (user_id,)) total_rewards = c.fetchone()[0] return { "num_referrals": num_referrals, "total_rewards": total_rewards }