python 资产扫描01

本地建立的三个文件：

Asset1.txt 用来保存扫描到的资产

Asset2.txt 用来导入给定的资产

Repeat.txt 保存重复的资产

程序的功能：

1.资产扫描，以 位置:资产 格式保存到Asset1文件中,如果资产重复，将重复的资产保存到Repeat文件中，按Q退出

2.资产比对，比对Asset1和Asset2 中的资产信息，输出两个文件的对称差集，即为有问题的资产

3.查看重复资产

4.清空所有数据

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2017/11/21 0021 16:01
# @Author  : ming

import re
import stat, os
import time

def Number():
    """
    :在第一行输出总资产数量信息
    """
    filename = "Asset1.txt"
    with open(filename) as myfile:
        lines = len(myfile.readlines())
        myfile.close()
        print "\033[5;37;44mTotal assets: [%s] in %s\033[0m" % (lines, filename)

def Repeat(asset):
    """
    :param asset:  资产信息参数
    :return: True代表资产重复，False代表资产没有重复
    """
    with open("Asset1.txt") as myfile:
        for i in myfile.readlines():
            asset2 = re.sub("-", "\-", asset)
            p = re.compile("%s$" % asset2)
            existence = p.findall(i)
            if len(existence) != 0:
                if asset == existence[0]:
                    return True
        return False

def SCAN():
    locate = "NONE"  # 默认位置
    ass = "NONE"  # 默认资产
    prompt = "This is prompt information"  # 默认提示语
    rock_list = []  # 当前机架资产列表

    os.chmod('Asset1.txt', stat.S_IWRITE)  # 取消只读
    flag = True
    with open("Asset1.txt", "r+") as file:
        file.seek(0, 2)
        while flag:
            os.system("cls")
            Number()  # 在第一行输出总资产数量信息
            print ("\033[1;33;44mCurrent rack locate: %s\033[0m" % locate)  # 输出当前所在机架
            print (prompt)  # 输出操作提示语
            print ("The current rack has scanned asset num: %d" % len(rock_list))  # 输出当前机架已经扫描的资产信息
            for i in rock_list:  # 打印当前机架已经扫描的资产
                print i
            scan = raw_input("Please scan[\"Q\" to quit]:")
            scan = scan.strip().upper()
            if re.search("R\d{1,2}C\d{1,2}", scan):  # 判断是否为位置信息
                locate = scan
                prompt = "\033[1;32;44mLocation update successful\033[0m"
                del rock_list[:]
                continue
            elif re.search("\w+-\w-\w+", scan):  # 判断是否为资产信息
                asset = scan
                repeat = Repeat(asset)  # 判断资产是否重复
                if repeat == True:
                    prompt = "\033[1;31;44mAsset information repetition!!!\033[0m"
                    repeat_data = ("%s:%s" % (locate, asset))  # 重复资产写入新文件
                    with open("Repeat.txt", "r+") as repete_file:
                        repete_file.seek(0, 2)
                        repete_file.write("%s\n" % repeat_data)
                        repete_file.flush()
                    continue
                elif repeat == False:
                    pass
            elif scan == "Q" or scan == "q":  # 判断是否退出
                break
            else:  # 输入错误
                prompt = "\033[1;31;44mInput error!!!\033[0m"
                continue
            data = ("%s:%s" % (locate, asset))
            if locate != "NONE" and asset != "NONE" and scan != "" and repeat != True:  # 写入文件
                file.write("%s\n" % data)
                file.flush()
                rock_list.append(asset)
                prompt = "\033[1;35;44mAsset update successful\033[0m"
            else:
                prompt = "\033[1;35;44mLocation or assets are empty\033[0m"
    os.chmod('Asset1.txt', stat.S_IREAD)  # 设置只读

def CHECK():
    """
    #比对
    :return: set1中存在set2中不存在；set2中存在set1中不存在
    """
    set1 = set([])
    set2 = set([])
    os.chmod('Asset1.txt', stat.S_IWRITE)  # 取消只读
    with open("Asset1.txt", "r+") as file1:
        for line1 in file1.readlines():
            set1.add(line1.strip())
    with open("Asset2.txt", "r+") as file2:
        for line2 in file2.readlines():
            set2.add(line2.strip())
    return set1.difference(set2), set2.difference(set1)
    os.chmod('Asset1.txt', stat.S_IREAD)  # 设置只读

def DUP():
    list3 = []
    set4 = set([])
    with open("Repeat.txt", "r+") as file1:
        for line1 in file1.readlines():
            list3.append(line1.strip())
        set4 = set(list3)
        return set4

def reset():
    os.chmod('Asset1.txt', stat.S_IWRITE)  # 取消只读
    sure = raw_input("Are you sure clear all data[Y/N]:")
    if sure == "y" or sure == "Y" or sure == "yes" or sure == "YES":
        with open("Asset1.txt", "w") as f1:
            pass
        with open("Asset2.txt", "w") as f2:
            pass
        with open("Repeat.txt", "w") as f3:
            pass
    os.chmod('Asset1.txt', stat.S_IREAD)  # 设置只读

def main():
    print "1.Scan assets"
    print "2.Assets verification"
    print "3.Duplicated assets"
    print "4.Clear all data"
    num = raw_input("please choise:")
    if num == "":
        SCAN()
    elif num == "":
        A, B = CHECK()
        print ("IN Asset1.txt NOT IN Asset2.txt")
        for i in A:
            print i
        print ("IN Asset2.txt NOT IN Asset1.txt")
        for i in B:
            print i
    elif num == "":
        C = DUP()
        for i in C:
            print i
    elif num == "":
        C = reset()
    raw_input("\nEnter to Quit!:")

if __name__ == '__main__':
    main()