python 實現網路商城,轉帳,存取款等功能的信用卡系統

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
import sys,os
import hashlib
import pickle
import time
import socketserver
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
from config import settings
from lib import modules
from lib.modules import *
 
class Myserver(socketserver.BaseRequestHandler):
 
 def md5(self,pwd):
  '''
  对密码进行加密
  :param pwd: 密码
  :return:
  '''
  hash=hashlib.md5(bytes('xx7',encoding='utf-8'))
  hash.update(bytes(pwd,encoding='utf-8'))
  return hash.hexdigest()
 
 
 def login(self,usrname,pwd,x):
  '''
  登陆
  :param usrname: 用户名
  :param pwd: 密码
  :return:是否登陆成功
  '''
  conn=self.request
  if x=='1':
   path_name_pwd=os.path.join(settings.ADMIN,usrname)
  else:
   mulu=os.path.join(settings.ALL_USERS,usrname)
   path_name_pwd=os.path.join(mulu,usrname+'name_pwd')
  s=pickle.load(open(path_name_pwd,'rb'))
  if usrname in s:
    if s[usrname]==self.md5(pwd):  #和加密后的密码进行比较
    return True
    else:
    return False
  else:
   return False
 
 
 def regist(self,usrname,pwd,x):
  '''
  注册
  :param usrname: 用户名
  :param pwd: 密码
  :return:是否注册成功
  '''
 
  conn=self.request
  if x=='1':
   mulu=os.path.join(settings.ADMIN,usrname)
  else:
   mulu=os.path.join(settings.ALL_USERS,usrname)
 
 
  if os.path.exists(mulu):
    return False
  else:
   os.mkdir(mulu)
   s={}
   s[usrname]=self.md5(pwd)
   path_name_pwd=os.path.join(mulu,usrname+'name_pwd')
   pickle.dump(s,open(path_name_pwd,'wb'))
   path_name_base=os.path.join(mulu,usrname+'name_base')
   pickle.dump([15000,{},0,0,0],open(path_name_base,'wb'))
   path_name_liushui=os.path.join(mulu,usrname+'name_liushui')
   os.mkdir(path_name_liushui)
   return True
 
 def user_identity_authentication(self,usrname,pwd,ret,x):
  '''
  判断注册和登陆，并展示用户的详细目录信息，支持cd和ls命令
  :return:
  '''
  conn=self.request
  if ret=='1':
   r=self.login(usrname,pwd,x)
   if r:
    conn.sendall(bytes('y',encoding='utf-8'))
   else:
    conn.sendall(bytes('n',encoding='utf-8'))
  elif ret=='2':
   # print(usrname,pwd)
   if x=='1':
    r=self.regist(usrname,pwd,x)
   else: #用户注册
    s=[0,1]
    pickle.dump(s,open(settings.A,'wb'))
    while True:
     ret=pickle.load(open(settings.A,'rb'))
     if ret[0]==0:
      time.sleep(30)
      continue
     elif ret[0]==1 or ret[0]==2:
      break #默认值已更改，银行管理员已操作
 
    if ret[0]==1: #如果管理员同意
     r=self.regist(usrname,pwd,x)
    else:
     r=0
    s=[0,0]
    pickle.dump(s,open(settings.A,'wb'))
   if r:
    conn.sendall(bytes('y',encoding='utf-8'))
   else:
    conn.sendall(bytes('n',encoding='utf-8'))
 def interactive(self,usrname): #进行交互
  conn=self.request
  while True:
   c=conn.recv(1024) #接收用户交互选项
   r=str(c,encoding='utf-8')
   mulu=os.path.join(settings.ALL_USERS,usrname)
   path_name_base=os.path.join(mulu,usrname+'name_base')
   s=pickle.load(open(path_name_base,'rb'))
 
   #打印账户信息
   obj=modules.Atm(s[0],s[1],s[2],s[3],s[4],usrname) #Atm对象
   a=obj.account_info() #接收账户信息
   conn.sendall(bytes(a,encoding='utf-8'))
 
   b=obj.ret_account_info()
 
 
 
   if r== '4':
    buy_obj=modules.buy(b[2],0,{})
    amount=buy_obj.main()
   elif r=='q':
    break
   else:
    s=conn.recv(1024)
    amount=str(s,encoding='utf-8')
 
   obj.transaction(r,amount)
 
 
  pass
 
 
 def handle(self):
  conn=self.request
  x=conn.recv(1024)
  x=str(x,encoding='utf-8')
  conn.sendall(bytes('收到用户类别',encoding='utf-8'))
  while True:
   if x=='1' or x=='2':
    b=conn.recv(1024)
    ret=str(b,encoding='utf-8')
    conn.sendall(bytes('b ok',encoding='utf-8'))
    c=conn.recv(1024)
    r=str(c,encoding='utf-8')
    usrname,pwd=r.split(',')
    print(usrname,pwd)
    self.user_identity_authentication(usrname,pwd,ret,x) #登陆或注册验证
    if x=='2':#普通用户身份验证成功后
     self.interactive(usrname)
 
     pass
 
    break
   elif x=='q':
    break
 
if __name__=='__main__':
 sever=socketserver.ThreadingTCPServer(('127.0.0.1',9999),Myserver)
 sever.serve_forever()