|
@@ -1,7 +1,7 @@
|
|
import pymysql
|
|
import pymysql
|
|
from config import set_config
|
|
from config import set_config
|
|
|
|
|
|
-config_ = set_config()
|
|
|
|
|
|
+config_, env = set_config()
|
|
|
|
|
|
|
|
|
|
class MysqlHelper(object):
|
|
class MysqlHelper(object):
|
|
@@ -34,5 +34,33 @@ class MysqlHelper(object):
|
|
conn.close()
|
|
conn.close()
|
|
return data
|
|
return data
|
|
|
|
|
|
|
|
+ def add_data(self, sql):
|
|
|
|
+ """
|
|
|
|
+ 新增数据
|
|
|
|
+ :param sql:
|
|
|
|
+ :return:
|
|
|
|
+ """
|
|
|
|
+ # 连接数据库
|
|
|
|
+ conn = pymysql.connect(**self.mysql_info)
|
|
|
|
+ # 创建游标
|
|
|
|
+ cursor = conn.cursor()
|
|
|
|
+ try:
|
|
|
|
+ # 执行SQL语句
|
|
|
|
+ cursor.execute(sql)
|
|
|
|
+ # 提交到数据库执行
|
|
|
|
+ conn.commit()
|
|
|
|
+ except Exception as e:
|
|
|
|
+ # 发生错误时回滚
|
|
|
|
+ conn.rollback()
|
|
|
|
+ # 关闭游标对象
|
|
|
|
+ cursor.close()
|
|
|
|
+ # 关闭数据库连接
|
|
|
|
+ conn.close()
|
|
|
|
+
|
|
|
|
|
|
|
|
+if __name__ == '__main__':
|
|
|
|
+ mysql_helper = MysqlHelper()
|
|
|
|
+ sql = "select * from hot_word;"
|
|
|
|
+ data = mysql_helper.get_data(sql=sql)
|
|
|
|
+ print(data)
|
|
|
|
|