import MySQLdb
# 打开数据库连接,如果存在表则删除,重新创建
def start():
db = MySQLdb.connect("127.0.0.1", "root", "xxx", "Ctf", charset='utf8' )
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# 使用execute方法执行SQL语句
cursor.execute("DROP TABLE IF EXISTS CTFSQL")
sqltable = """CREATE TABLE CTFSQL(
ID INT NOT NULL,
TIMU_NAME VARCHAR(100) NOT NULL,
TEAM_NAME VARCHAR(100) NOT NULL,
PORT INT NOT NULL,
COMPUTERIP VARCHAR(100) NOT NULL,
FLAG VARCHAR(100)
)"""
cursor.execute(sqltable)
db.close()
#传递参数插入数据
def connectCtf(ID,TIMU_NAME,TEAM_NAME,PORT,COMPUTERIP):
db = MySQLdb.connect("127.0.0.1", "root", "xxx", "Ctf", charset='utf8')
#sqldata = """INSERT INTO CTFSQL(ID,TIMU_NAME,TEAM_NAME, PORT)VALUES (1,"Webtimu", "sunjikui",8 000)"""
cursor = db.cursor()
sqldata="INSERT INTO CTFSQL (ID,TIMU_NAME,TEAM_NAME, PORT,COMPUTERIP) VALUES(\'%s\',\'%s\',\'%s \',\'%s\',\'%s\')" %(ID,TIMU_NAME,TEAM_NAME, PORT,COMPUTERIP)
print sqldata
try:
# 执行sql语句
cursor.execute(sqldata)
# 提交到数据库执行
db.commit()
print "success"
except:
# 发生错误时回滚
db.rollback()
print "fail"
# 关闭数据库连接
# 关闭数据库连接
db.close()
# 传递参数更新数据
def UpdateCtf(FLAG,COMPUTERIP):
db = MySQLdb.connect("127.0.0.1", "root", "xxx", "Ctf", charset='utf8' )
# 使用cursor()方法获取操作游标
cursor = db.cursor()
sqldata="update CTFSQL set FlAG='%s' where COMPUTERIP='%s'" %(FLAG,COMPUTERIP)
print sqldata
try:
# 执行sql语句
cursor.execute(sqldata)
# 提交到数据库执行
db.commit()
print "success"
except:
# 发生错误时回滚
db.rollback()
print "fail"
# 关闭数据库连接
# 关闭数据库连接
db.close()
if __name__ == "__main__":
connectCtf()
UpdateCtf()