Python代码:从数组中随机获取不重复ID并写入数据库
import requests import json import re import pymysql import random
db = pymysql.connect(host='103.214.173.233', port=3306, user='caoliu', passwd='xX300400', db='caoliu') cursor = db.cursor()
假设html变量包含了你要提取的HTML内容
以下代码示例:
html = '链接1链接2'
pattern = re.compile(r'_blank' id='t(.*?)'>') ids = pattern.findall(html) print(ids)
def check_duplicate(id_value): query = f'SELECT * FROM huifu WHERE id = '{id_value}'' cursor.execute(query) result = cursor.fetchone() return result is not None
def generate_unique_id(ids): while True: random_id = random.choice(ids) if not check_duplicate(random_id): return random_id
random_id = generate_unique_id(ids) print(f'Unique ID: {random_id}')
Write the unique ID to the database
insert_query = f'INSERT INTO huifu (id) VALUES ('{random_id}')' cursor.execute(insert_query) db.commit()
原文地址: https://www.cveoy.top/t/topic/cg4o 著作权归作者所有。请勿转载和采集!