import pymysql
#1.连接服务器
conn=pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123456',
database='公告'
)
#2.通过连接拿到游标对象
#默认的游标返回的是元祖类型,不方便使用,需要更换字典类型的游标
c=conn.cursor(pymysql.cursors.DictCursor)
is_login=[None]
def login_auth(func):
def wrapper(*args, **kwargs):
while not is_login[0]:
print('请先登录')
login(c,conn)
res=func(*args,**kwargs)
return res
return wrapper
#用户注册
def register(c,conn):
while True:
print('欢迎来到注册')
#输入用户信息
username=input('输入你的用户名:')
pwd=input('输入你的密码:')
#3.执行sql
sql='select * from user where username=%s'
if c.execute(sql,(username,)):
print('用户名已存在,请重新输入!')
continue
else:
sql2='insert into user(username,pwd) values(%s,%s)'
if c.execute(sql2,(username,pwd)):
print('注册成功!')
conn.commit()
break
else:
print('注册失败')
continue
#用户登录
def login(c,conn):
global is_login
if is_login[0]:
print('已经登录了不用再登录')
return
while True:
print('欢迎来到登录')
#输入用户信息
username=input('输入你的用户名:')
pwd=input('输入你的密码:')
sql='select * from user where username=%s and pwd=%s'
rec=c.execute(sql,(username,pwd))
if c.fetchall():
print('登录成功')
conn.commit()
is_login[0]=1
break
else:
print('账号密码错误')
#发布公告
@login_auth
def post_notice(c,conn):
print('欢迎来到发布公告')
notice_titile=input('请输入你要发布的公告标题:')
notice_content= input('请输入你要发布的公告内容:')
#执行sql
sql='insert into notice(titile,content) values(%s,%s)'
c.execute(sql,(notice_titile,notice_content))
print('公告发布成功')
conn.commit()
#查看公告
@login_auth
def show_notice(c,conn):
print('欢迎来到查看公告')
sql='select * from notice'
c.execute(sql)
print(c.fetchall())
conn.commit()
func_list={
1:register,
2:login,
3:post_notice,
4:show_notice
}
func_list_view='''
1:register,
2:login,
3:post_notice,
4:show_notice
5:quit
'''
while True:
print(func_list_view)
choice=input('请选择功能:')
if choice=='5':
break
else:
choice=int(choice)
func_list[choice](c,conn)