# -*- coding: UTF-8 -*-
import os
from pathlib import Path

'''
A對比B   將A與B的交集或并集等結果輸出到一個文件中
注意格式必須一致  按行讀取  必要時 在EXCEL統一下格式
intersection
'''

pathA = Path(r'C:\Users\EDZ\Desktop\A.txt')#獲取A文件路徑  A和B需要手動先創建txt
pathB = Path(r'C:\Users\EDZ\Desktop\B.txt')  # 獲取B文件路徑
pathresult = Path(r'C:\Users\EDZ\Desktop\result.txt')#結果路徑  result.txt是自動生成

'''
讀取文件A 
'''
#目錄臨時限制死 后改為輸入形式并抓異常
def read_file_A():

    with open(pathA,mode='r', encoding='utf-8') as f: #2.讀取文件
         content = f.readlines()
         Aset = set(content)    #3.將文件按行存入setA中 注意空格與換行符
         return Aset
         #print(Aset)


'''
讀取文件B
'''

def read_file_B():

    #目錄臨時限制死 后改為輸入形式并抓異常
    with open(pathB,mode='r', encoding='utf-8') as f:  # 2.讀取文件
        content = f.readlines()
        Bset = set(content)  # 3.將文件按行存入setA中 注意空格與換行符
        return Bset
        # print(Bset)


'''
比較
'''
def compare_file():

    A_Set = read_file_A()
    B_Set = read_file_B()
    output_comparefile_result(A_Set, B_Set)
    print('完成')

'''
輸出文件到指定位置
'''
#目錄臨時限制死
def output_comparefile_result(A_set,B_set):

    input_menu = input('菜單:輸入1或2或3\n1.查出A和B相同的\n2.查出A存在B不存在的\n3.查出B存在A不存在的\n')
    with open(pathresult, 'w', encoding="utf-8") as f:
        if(input_menu=='1'):
            str=A_set.intersection(B_set) #查出相同的
        elif(input_menu=='2'):
            str=A_set.difference(B_set) #查出不同的
        elif(input_menu=='3'):
            str = B_set.difference(A_set)  # 查出不同的
        else:
           raise  ValueError('Error，發生異常，只能輸入1、2、3')

        for x in str:
            f.write(x)

#程序入口

if __name__ == '__main__':

    try:
        compare_file()

    except Exception as e:
        print(e)