1

2

3

4

import os  # 应用文件操作

import hashlib  # 文件对比操作

import logging  # 日志函数操作

import sys  # 系统应用操作

1

2

3

logger = logging.getLogger('系统文件去重')

logging.basicConfig(format='%(asctime)s %(levelname)-8s: %(message)s')

logger.setLevel(logging.DEBUG)

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

diretory = input('请输入需要整理的文件目录: ')  # 去重的文件夹路径

if os.path.isdir(diretory):

    logger.info('当前目录[' + diretory + ']校验成功！')

    md5s = []

    for file_path, dir_names, file_names in os.walk(r'' + diretory):

        for file_name in file_names:

            try:

                file_name_path = os.path.join(file_path, file_name)

                logger.info('当前比对路径: '+ file_name_path)

                md5 = hashlib.md5()

                file = open(file_name_path, "rb")

                md5.update(file.read())

                file.close()

                md5_value = md5.hexdigest()

                if md5_value in md5s:

                    os.remove(file_name_path)

                    logger.info('[' + file_name_path + ']出现重复已经移除！')

                else:

                    md5s.append(md5_value)

            except:

                logger.error('[' + file_name_path + ']对比发生异常，执行下一个！')

else:

    logger.error('输入的文件夹或者目录不存在！')

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

#!/usr/bin/env python

#-*- coding:utf-8 -*-

"""

功能：

    登录验证模块

详细说明：

    1.密码文件为passwd

    2.passwd未创建或丢失，会提示：密码文件不存在，建议重新注册！！

    3.未注册用户登录会提示：用户名不存在，请您先进行注册！

    4.已注册用户登录时，忘记密码，尝试3次后密码还不正确则退出验证，等一会儿则可以重新登录

    5.作为装饰器进行登录验证

"""

import json

import hashlib

import os

pwd = os.getcwd()

fileName = os.path.join(pwd,"passwd")

#将明文密码通过md5进行加密,返回一个加密后的md5的值

def calc_md5(passwd):

    md5 = hashlib.md5("haliluya")

    md5.update(passwd)

    ret = md5.hexdigest()

    return ret

#新用户注册模块

def register():

    #判断密码文件passwd是否存在，存在则载入列表，不存在就重新创建一个空字典

    if os.path.exists(fileName):

        #载入用户列表，数据结构为字典，k=userName,v=passwdMd5

        with open("passwd", "r+") as loadsFn:

            userDB = json.loads(loadsFn.read())

    else:

        userDB = {}

    #让用户输入用户名

    userName = raw_input("姓名：")

    #标志位：控制循环跳出

    flag = True

    while flag:

        #用户注册时，需输入两次密码

        passwd1 = raw_input("密码：")

        passwd2 = raw_input("确认密码：")

        #如果两次密码不一致，则不执行下一步，再次输入密码并进行确认

        if not passwd1 == passwd2:

            continue

        else:

            #两次输入密码一致，标志位置为False，下次跳出循环

            flag = False

        #调用calc_md5函数将明文密码转为对应的md5值，用于保存

        passwdMd5 = calc_md5(passwd1)

    #将用户名与密码对应存入字典userDB中

    userDB[userName] = passwdMd5

    #将用户名和密码存入文件

    with open(fileName, "w") as dumpFn:

        dumpFn.write(json.dumps(userDB))

#用户登录验证,装饰器

def login(func):

    def decorater(*args,**kwargs):

        #判断passwd文件是否存在，存在则载入userDB（用户：密码），否则就重新注册新的passwd文件并返回

        if os.path.exists(fileName):

            with open("passwd", "r+") as loadsFn:

                userDB = json.loads(loadsFn.read())

        else:

            print "密码文件不存在，建议重新注册！！"

            register()

            return

        name = raw_input("用户名：")

        #用户名是否存在，存在就继续输入密码，不存在则进行注册

        if name in userDB.keys():

            flag = True

            counter = 0

            #循环输入密码，密码正确，flag=False（下次直接跳出循环）并执行函数，密码错误则允许尝试3次，超过3次验证失败，退出验证

            while flag:

                passwd = raw_input("密码：")

                passwdMd5 = calc_md5(passwd)

                if passwdMd5 == userDB[name]:

                    flag = False

                    func(*args,**kwargs)

                elif counter > 2:

                    print "您已经尝试了3次，请过会儿再试！！"

                    return

                else:

                    counter += 1

        else:

            print "用户名不存在，请您先进行注册！"

            register()

    return decorater

if __name__ == "__main__":

    @login

    def hello():

        print "Hello world!"

    hello()