gameui/ocr
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

簡化跳行邏輯

Browse Source
This commit is contained in:
satori
2024-11-19 15:27:40 +08:00
parent ef0ef48b87
commit 4465f1f9f0
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pp.py
View File

@@ -81,10 +81,7 @@ JP = PaddleOCR(use_angle_cls=True, lang="japan")
KR = PaddleOCR(use_angle_cls=True, lang="korean") KR = PaddleOCR(use_angle_cls=True, lang="korean")
RU = PaddleOCR(use_angle_cls=True, lang="ru") RU = PaddleOCR(use_angle_cls=True, lang="ru")
offset=1500 def process_images(conn, offset=0) -> int:
def process_images(conn):
global offset
with conn.cursor(pymysql.cursors.SSCursor) as cursor: with conn.cursor(pymysql.cursors.SSCursor) as cursor:
cursor.execute("SELECT id, content FROM web_images WHERE text='' AND article_category_top_id=22 LIMIT 100 OFFSET %s", (offset,)) cursor.execute("SELECT id, content FROM web_images WHERE text='' AND article_category_top_id=22 LIMIT 100 OFFSET %s", (offset,))
for id, content in cursor.fetchall(): for id, content in cursor.fetchall():
@@ -142,12 +139,13 @@ def process_images(conn):
data = json.dumps(data, ensure_ascii=False, cls=MyEncoder) data = json.dumps(data, ensure_ascii=False, cls=MyEncoder)
cursor.execute("UPDATE web_images SET text = %s WHERE id = %s", (data, id)) cursor.execute("UPDATE web_images SET text = %s WHERE id = %s", (data, id))
conn.commit() conn.commit()
offset+=100 return offset+100
def main(): def main():
conn = connect_to_mysql() conn = connect_to_mysql()
offset = 1500
while True: while True:
process_images(conn) offset = process_images(conn, offset)
time.sleep(0) time.sleep(0)