98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
from sshtunnel import SSHTunnelForwarder
|
||
import psycopg2
|
||
|
||
class PGSQL():
|
||
def __init__(self):
|
||
ssh_host = 'ops.yimaogo.com'
|
||
ssh_username = 'yimaogo_test'
|
||
ssh_pwd = 'yimaogo!test@1234'
|
||
ssh_port = 22
|
||
db_host = '10.0.16.24'
|
||
db_port = 5432
|
||
db_dbname = 'yimaogo_test'
|
||
db_user = 'yimaogo_test'
|
||
db_password = 'yimaogo!test@1234'
|
||
self.ssh_server = SSHTunnelForwarder(
|
||
ssh_address_or_host=(ssh_host, ssh_port),
|
||
ssh_username=ssh_username,
|
||
ssh_password=ssh_pwd,
|
||
remote_bind_address=(db_host, db_port)
|
||
)
|
||
self.ssh_server.daemon_forward_servers = False
|
||
self.ssh_server.start()
|
||
self.conn = psycopg2.connect(
|
||
database=db_dbname,
|
||
user=db_user,
|
||
password=db_password,
|
||
host='127.0.0.1',
|
||
port=self.ssh_server.local_bind_port
|
||
)
|
||
self.cursor = self.conn.cursor()
|
||
|
||
def close(self):
|
||
self.cursor.close()
|
||
self.conn.close()
|
||
self.ssh_server.close()
|
||
|
||
def select_data(self, SQL):
|
||
self.cursor.execute(SQL)
|
||
# 获取查询结果
|
||
# rows = self.cursor.fetchone()
|
||
# print(rows)
|
||
rows = self.cursor.fetchall()
|
||
# print(rows)
|
||
# for row in rows:
|
||
# print(row)
|
||
return rows
|
||
|
||
def delete_data(self, SQL):
|
||
self.cursor.execute(SQL)
|
||
self.conn.commit()
|
||
|
||
def del_sample_activity_data(self, IDList=None):
|
||
import time
|
||
print('---开始删除派样活动自动化测试脏数据')
|
||
for SampActivityID in IDList:
|
||
# for SampActivityID in range(120, 140):
|
||
print(f'---待删除ID为：{SampActivityID}')
|
||
selsqlstr = f'SELECT * FROM public.sample_activity WHERE id={SampActivityID};'
|
||
delsqlstr = [
|
||
f"DELETE FROM public.sample_activity_gift WHERE activity_id={SampActivityID};",
|
||
f"DELETE FROM public.sample_record WHERE activity_id={SampActivityID};",
|
||
f"DELETE FROM public.sample_activity_store WHERE activity_id={SampActivityID};"
|
||
f"DELETE FROM public.sample_activity WHERE id={SampActivityID};"
|
||
]
|
||
for s in delsqlstr:
|
||
time.sleep(0.5)
|
||
self.delete_data(s)
|
||
time.sleep(0.5)
|
||
self.select_data(selsqlstr)
|
||
self.close()
|
||
print("---删除 派样活动 脏数据结束！\n")
|
||
|
||
def del_coupon_activity_data(self, IDList=None):
|
||
try:
|
||
import time
|
||
print('---开始删除 优惠券活动 自动化测试脏数据')
|
||
for ActivityID in IDList:
|
||
# for ActivityID in range(251, 264):
|
||
print(f'---待删除ID为：{ActivityID}')
|
||
selsqlstr = f'SELECT * FROM public.coupon_setting WHERE id={ActivityID};'
|
||
delsqlstr = [
|
||
f"DELETE FROM public.coupon_setting_distribution WHERE coupon_setting_id={ActivityID};",
|
||
f"DELETE FROM public.coupon_setting WHERE id={ActivityID};",
|
||
]
|
||
for s in delsqlstr:
|
||
time.sleep(0.5)
|
||
self.delete_data(s)
|
||
time.sleep(0.5)
|
||
self.select_data(selsqlstr)
|
||
self.close()
|
||
print("---删除派样活动脏数据结束！\n")
|
||
return "---[Successed]"
|
||
except Exception as e:
|
||
print(f"---删除派样活动脏数据失败: {e}\n")
|
||
return e
|
||
|