diff --git a/requirements.txt b/requirements.txt index a46aa9c..05807aa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,5 @@ pytz==2021.1 filetype==1.0.7 requests[socks] tqdm==4.61.1 -retry==0.9.2 \ No newline at end of file +retry==0.9.2 +pymysql \ No newline at end of file diff --git a/yyetsweb/migration/dump/convert_to_mysql.py b/yyetsweb/migration/dump/convert_to_mysql.py new file mode 100644 index 0000000..77f9093 --- /dev/null +++ b/yyetsweb/migration/dump/convert_to_mysql.py @@ -0,0 +1,47 @@ +#!/usr/local/bin/python3 +# coding: utf-8 + +# YYeTsBot - convert_to_sqlite.py +# 6/17/21 12:41 +# + +__author__ = "Benny " + +import json +import pymysql + +import pymongo + +mongo = pymongo.MongoClient() +yyets = mongo["zimuzu"]["yyets"] + +con = pymysql.connect(host="mysql", user="root", password="root", charset="utf8mb4") +cur = con.cursor() +cur.execute("create database zimuzu") +cur.execute("use zimuzu") +TABLE_SQL = """ +CREATE TABLE IF NOT EXISTS yyets +( + id int, + cnname text, + enname text, + aliasname text, + views int, + data longtext +); +""" +cur.execute(TABLE_SQL) + +INSERT_SQL = """ +INSERT INTO yyets VALUES (%s, %s, %s, %s ,%s, %s); +""" +for resource in yyets.find(projection={"_id": False}): + resource_id = resource["data"]["info"]["id"] + cnname = resource["data"]["info"]["cnname"] + enname = resource["data"]["info"]["enname"] + aliasname = resource["data"]["info"]["aliasname"] + views = resource["data"]["info"]["views"] + cur.execute(INSERT_SQL, (resource_id, cnname, enname, aliasname, views, json.dumps(resource, ensure_ascii=False))) + +con.commit() +con.close() diff --git a/yyetsweb/migration/convert_to_sqlite.py b/yyetsweb/migration/dump/convert_to_sqlite.py similarity index 100% rename from yyetsweb/migration/convert_to_sqlite.py rename to yyetsweb/migration/dump/convert_to_sqlite.py