import os
BASE_DIR = os.path.dirname(os.path.dirname(__file__)) #配置文件的上層目錄
DB_DIR=os.path.join(BASE_DIR,'db')    #數(shù)據(jù)文件夾
ADMIN=os.path.join(DB_DIR,'admin')
ALL_USERS=os.path.join(DB_DIR,'allusrs')
A=os.path.join(BASE_DIR,'db','s')
LOG=os.path.join(BASE_DIR,'log')
TRANSACTION={
  'repay':{'action':'plus','interest':0}, #還款
  'withdraw':{'action':'minus','interest':0.05},#取現(xiàn)
  'transfer':{'action':'minus','interest':0.05},#轉(zhuǎn)賬
  'consume':{'action':'minus','interest':0},#消費(fèi)
  'saving':{'action':'plus','interest':0} #存款
}

class buy:
  goods=[
      {"name": "電腦", "price": 1999},
      {"name": "鼠標(biāo)", "price": 10},
      {"name": "游艇", "price": 20},
      {"name": "美女", "price": 998},
    ]
  def __init__(self,money,consumption,shopping_cart,):
    self.money=money
    self.consumption=consumption
    self.shopping_cart=shopping_cart
  def gouwu(self): #購物模塊
    print('您的當(dāng)前余額為：%d' %self.money)
    num=int(input('請(qǐng)輸入商品序號(hào)：'))
    num-=1
    if self.goods[num]["name"] in self.shopping_cart.keys():        #goods[num]["name"]取商品名
      self.shopping_cart[self.goods[num]["name"]]['n']+=1         #商品數(shù)量+1
    else:
      self.shopping_cart[self.goods[num]["name"]]={"price":self.goods[num]["price"],'n':1,} # 創(chuàng)建購物車字典  {keys{"price"：價(jià)格,數(shù)量：1}}
    self.money-=self.shopping_cart[self.goods[num]["name"]]["price"]*self.shopping_cart[self.goods[num]["name"]]['n']    #單價(jià)*數(shù)量
    self.consumption+=self.shopping_cart[self.goods[num]["name"]]["price"]*self.shopping_cart[self.goods[num]["name"]]['n']
  def yichu(self): #移除購物車模塊
    c=int(input(' 請(qǐng)輸入0/1選擇是否移除購物車商品, 移除請(qǐng)輸入1:'))
    if c==1:
      e=int(input(' 請(qǐng)輸入要移除的商品序號(hào):'))
      d=self.goods[e-1]
      if d in self.shopping_cart.keys():       #判斷要移除的商品是否在購物車內(nèi)
        self.shopping_cart.remove(d)     #移除商品
        self.money=self.money+self.goods[self.goods.index(d)]["price"]       #余額增加
        self.consumption=self.consumption-self.goods[self.goods.index(d)]["price"] #消費(fèi)總額減少
      else:
        print('商品不存在')
  def chongzhi(self): #充值模塊
    pay=int(input('請(qǐng)輸入充值金額'))
    self.money=self.money+pay
    print('您的當(dāng)前余額為：%d' % self.money)  #顯示當(dāng)前余額
  def main(self):
    print('商品清單：')
    for m,n in enumerate(self.goods,1):
      print(m)
      for v in n.values():
        print(v)
      print('=============')
    #消費(fèi)總額清零
    self.consumption=0
    buy=True #定義默認(rèn)一直購物
    while buy:
      price=0 #定義初始價(jià)格
      b=1 #定義默認(rèn)不退出購物或充值狀態(tài)
      if self.money>=price:
    #消費(fèi)模塊；金錢大于貨物價(jià)格時(shí)，才能開始購物
        while self.money>=price:
    #計(jì)價(jià)模塊，有錢就可以一直購物
          self.gouwu()
    #移除購物車商品模塊
          self.yichu()
          if self.money>=0:
            print('您的當(dāng)前余額為：%d' %self.money)     #顯示當(dāng)前余額
            b=int(input(' 請(qǐng)輸入0/1選擇是否繼續(xù)購物, 購物請(qǐng)輸入1:'))
            if b==0:    #
              break  #退出計(jì)價(jià)模塊
        if b==0:      #如果不購物
          break     #不購物退出整個(gè)購物程序
    #充值模塊
      else:
        while self.money<price:     #金錢不足，可多次充錢，直到能買得起貨物
          a=int(input(' 您的余額不足，請(qǐng)輸入0/1選擇是否充值，充值請(qǐng)輸入1:'))
          if a==1:
            self.chongzhi()
          else:
            break      #退出充值模塊
          if a==0:
            break      #不充值退出程序
    #打印購物車商品名、商品價(jià)格、消費(fèi)總額、余額
    print('您的消費(fèi)清單為：')
    for m,n in self.shopping_cart.items():
      print(m,n['price'],n['n'])
          #打印消費(fèi)清單
      print('=============')
    print('您的當(dāng)前余額為：%d,您的消費(fèi)總額為：%d' % (self.money,self.consumption) )      #打印消費(fèi)總額
    return self.consumption

class Atm:
  credit=15000 #信用卡額度
  def __init__(self,balance,debt,remaining_debt,interest,saving,id):
    self.id=id       #信用卡id
    self.balance=balance   #信用卡可用金額
    self.debt=debt      #總欠款
    self.remaining_debt=remaining_debt #剩余欠款
    self.interest=interest   #手續(xù)費(fèi)
    self.saving=saving   #存款
    self.now_time=time.strftime("%Y-%m-%d %H:%M:%S")
    self.now_data=time.strftime("%Y-%m")
    self.struct_time=time.gmtime(time.time())
    if self.struct_time.tm_mday>22:
      self.now_data=self.struct_time.tm_year+'-'+str(int(self.struct_time.tm_mon)+1)
  def account_info(self):#打印賬戶信息
    return '賬戶id%s 信用卡額度%s;信用卡可用金額%s;剩余欠款%s;'%(self.id,self.credit,self.balance,self.remaining_debt,)
  def ret_account_info(self):
    return [self.id,self.credit,self.balance,self.debt,self.remaining_debt,self.interest]
  def repay(self,amount):#還款
    self.handel_money('repay',amount)
  def withdraw(self,amount): #取現(xiàn)
    self.handel_money('withdraw',amount)
  def transfer(self,amount): #轉(zhuǎn)賬
    self.handel_money('transfer',amount)
  def consume(self,amount): #消費(fèi)
    self.handel_money('consume',amount)
  def saves(self,amount):
    self.handel_money('saving',amount)
  def transaction(self,a,amount):
    dic={
      '1':self.repay,
      '2':self.withdraw,
      '3':self.transfer,
      '4':self.consume,
      '5':self.saves
    }
    print("debug: a:",type(a),"amount:",type(amount))
    print(a)
    print(dic[a])
    print(dic["5"])
    dic[a](amount)
    print("end debug")
  def handel_money(self,transaction,amount): #交易類型，
    amount=int(amount)
    interest=amount*settings.TRANSACTION[transaction]['interest'] #手續(xù)費(fèi)計(jì)算
    if settings.TRANSACTION[transaction]['action']=='plus':
      if amount<=self.remaining_debt:
        self.remaining_debt-=amount
        self.balance+=amount
      else:
        self.balance+=self.remaining_debt
        self.remaining_debt=0
        self.saving+=amount-self.remaining_debt
    else:
      if self.saving<amount:
        self.saving=0
        a=amount-self.saving
        self.balance-=a+interest-self.saving
        # self.debt+=amount+interest
        self.remaining_debt+=a+interest
    a='time:%s id:%s transaction: %s amount:%s interest %s \n'%(self.now_time,self.id,transaction,amount,interest)
    print(a)
    mulu=os.path.join(settings.ALL_USERS,self.id)
    path_name_liushui=os.path.join(mulu,str(self.id)+'name_liushui',str(self.now_data))
    with open(path_name_liushui,'a')as f:     #記錄流水信息
      f.write(a)
    s=[self.balance,self.debt,self.remaining_debt,self.interest,self.saving,]   #更新基本信息
    path_name_base=os.path.join(mulu,str(self.id)+'name_base')
    pickle.dump(s,open(path_name_base,'wb'))

#!/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):
    '''
    對(duì)密碼進(jìn)行加密
    :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):    #和加密后的密碼進(jìn)行比較
        return True
       else:
        return False
    else:
      return False
  def regist(self,usrname,pwd,x):
    '''
    注冊(cè)
    :param usrname: 用戶名
    :param pwd: 密碼
    :return:是否注冊(cè)成功
    '''
    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):
    '''
    判斷注冊(cè)和登陸，并展示用戶的詳細(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:  #用戶注冊(cè)
        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 #默認(rèn)值已更改，銀行管理員已操作
        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): #進(jìn)行交互
    conn=self.request
    while True:
      c=conn.recv(1024) #接收用戶交互選項(xiàng)
      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對(duì)象
      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) #登陸或注冊(cè)驗(yàn)證
        if x=='2':#普通用戶身份驗(yàn)證成功后
          self.interactive(usrname)
          pass
        break
      elif x=='q':
        break
if __name__=='__main__':
  sever=socketserver.ThreadingTCPServer(('127.0.0.1',9999),Myserver)
  sever.serve_forever()

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
本程序作為用戶或銀行管理員的入口，其中c=1代表銀行管理員，c=2代表普通用戶
'''
import pickle
import sys
import time
import os
import socket
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
from config import settings
from lib import *
from lib.modules import *
def login(usrname,pwd):
  '''
  登陸
  :param usrname:用戶名
  :param pwd:密碼
  :return:是否登陸成功
  '''
  obj.sendall(bytes(usrname+','+pwd,encoding='utf-8'))
  ret=obj.recv(1024)
  r=str(ret,encoding='utf-8')
  if r=='y':
    return 1
  else:
    return 0
def regist(usrname,pwd,x):
  '''
  注冊(cè)
  :param usrname:用戶名
  :param pwd:密碼
  :return:是否注冊(cè)成功
  '''
  obj.sendall(bytes(usrname+','+pwd,encoding='utf-8'))
  ret=obj.recv(1024)
  r=str(ret,encoding='utf-8')
  if r=='y':
    return 1
  else:
    return 0
def user_identity_authentication(usrname,pwd,x):
  '''
  選擇登陸或注冊(cè)，展示用戶的詳細(xì)目錄信息，支持cd和ls命令
  :return:
  '''
  a=input('請(qǐng)選擇1.登陸 2.注冊(cè)')
  obj.sendall(bytes(a,encoding='utf-8'))
  obj.recv(1024)
  if a=='1':
    ret=login(usrname,pwd)
    if ret:
      print('登陸成功')
      return 1
    else:
      print('用戶名或密碼錯(cuò)誤')
      return 0
  elif a=='2':
    ret=regist(usrname,pwd,x)
    if ret:
      print('注冊(cè)成功')
      return 1
    else:
      print('用戶名已存在或銀行管理員拒絕')
      return 0
def main(x):
  usrname=input('請(qǐng)輸入用戶名')
  pwd=input('請(qǐng)輸入密碼')
  if user_identity_authentication(usrname,pwd,x): #如果驗(yàn)證身份成功
    if x=='1':  #處理用戶注冊(cè)信息
      while True:
        s=pickle.load(open(settings.A,'rb'))
        if s[1]==0:
          time.sleep(30)
          continue
        elif s[1]==1:
          while True:
            a=input('用戶請(qǐng)求注冊(cè)，輸入1同意，2拒絕')
            if a=='1':
              s=[1,0]
              pickle.dump(s,open(settings.A,'wb'))
              break
            elif a=='2':
              s=[2,0]
              pickle.dump(s,open(settings.A,'wb'))
              break
            else:
              print('輸入有誤')
          break
    else: #普通用戶登陸后
      interactive() #進(jìn)行交互
def interactive():
  while True:
    a=input('請(qǐng)選擇 1.還款 2.取現(xiàn) 3.轉(zhuǎn)賬 4.消費(fèi) 5.存錢 q退出')
    obj.sendall(bytes(a,encoding='utf-8'))
    r=obj.recv(1024) #接收賬戶信息
    ret=str(r,encoding='utf-8')
    print(ret)
    if a !='4'and a !='q':
      b=input('請(qǐng)輸入金額')
      obj.sendall(bytes(b,encoding='utf-8'))
    elif a=='q':
      break
obj=socket.socket() #創(chuàng)建客戶端socket對(duì)象
obj.connect(('127.0.0.1',9999))
while True:
  x=input('請(qǐng)選擇1.銀行管理員 2.用戶 q、退出')
  obj.sendall(bytes(x,encoding='utf-8'))
  obj.recv(1024)  #確認(rèn)收到用戶類別
  if x=='1' or x=='2':
    main(x)
    break
  elif x=='q':
    break
  else:
    print('輸入有誤請(qǐng)重新輸入')
obj.close()

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import os,sys
import json,pickle
import time
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
from config import settings
def main():
  card_list = os.listdir(settings.ALL_USERS)
  for card in card_list:
    basic_info = pickle.load(open(os.path.join(settings.ALL_USERS, card, card+'name_base')))
    struct_time = time.localtime()
    # 循環(huán)賬單列表，為每月的欠款計(jì)息。并寫入到當(dāng)月賬單中
    for item in basic_info['debt']:
      interest = item['total_debt'] * 0.0005
      if basic_info[4] >= interest:
        basic_info[4] -= interest
      else:
        temp = interest - basic_info[4]
        basic_info[4]=0
        basic_info[0] -= temp
        pickle.dump(
            basic_info,
            open(os.path.join(settings.ALL_USERS, card, card+'name_base'),'w')
       )
    # 如果當(dāng)前等于10號(hào)(9號(hào)之前)
    #  當(dāng)前余額為負(fù)值，則將值添加到賬單列表中，開始計(jì)息，同時(shí)，本月可用額度恢復(fù)。
    date = time.strftime("%Y-%m-%d")
    if struct_time.tm_mday == 11 and basic_info[2]>0:
      dic = {'date': date,
          "total_debt": basic_info[2],
          "balance_debt": basic_info[2],
          }
      basic_info[1].append(dic)
      # 恢復(fù)可用額度
      basic_info[0] = 15000
    pickle.dump(
      basic_info,
      open(os.path.join(settings.ALL_USERS, card, card+'name_base'),'w')
       )
def run():
  main()