代码审计双十一优惠活动

# 示例：使用Redis缓存热门商品信息
import redis

r = redis.Redis(host='localhost', port=6379, db=0)

def get_product_info(product_id):
    info = r.get(f'product:{product_id}')
    if info is None:
        info = fetch_from_db(product_id)  # 假设这是从数据库获取信息的函数
        r.setex(f'product:{product_id}', 3600, info)  # 缓存1小时
    return info

# 示例：使用参数化查询防止SQL注入
import sqlite3

def get_user_info(user_id):
    conn = sqlite3.connect('example.db')
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM users WHERE id=?", (user_id,))
    user_info = cursor.fetchone()
    conn.close()
    return user_info

# 示例：编写单元测试用例
import unittest

class PromotionTestCase(unittest.TestCase):
    def test_discount_calculation(self):
        self.assertEqual(calculate_discount(100, 0.1), 10)
        self.assertEqual(calculate_discount(50, 0.2), 10)
        self.assertEqual(calculate_discount(0, 0.1), 0)

if __name__ == '__main__':
    unittest.main()