Serialization_with_Python/main.py at master · kimkyury/Serialization_with_Python · GitHub

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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
from factory.object import *  #얘를 통해서 contol도 상속받음
from factory.jsonControl import *
from factory.pickleControl import *
from connectDB.postgre import *
import pickle

def select_format() : #형식을 지정하는 함수
    form_flag = input("엑셀로 진행:1, Json으로 진행:2, Pickle로 진행:3, DB로 진행:4 \n선택해주세요: ")
    return form_flag


if __name__ == "__main__":  # 프로젝트 실행과 함께 시작

    act_flag = int(input("새 파일 생성하기:1, 이전 파일 추가하기:2, 파일 확인하기:3 \n선택해주세요: "))
    form_flag = int(select_format())

    #새파일 제작
    if ((act_flag ==1) and (form_flag ==   1)):   #엑셀로
        IPmans = faker_generator()  # Fake 인간 생성/
        wb = xl.create_excel() # 기본엑셀파일 생성.  (def from excelControl.py)import object.py에서 control.py를 끌고왔음
        xl.set_Ipmans(wb['data'], IPmans) # 만든 xlsx에 fake담기 (def from excelControl.py)
        xl.save_excel(wb) #만든 엑셀파일 지정경로로 저장
        print('생성완료')

    elif ((act_flag ==1) and (form_flag == 2)):   # 새파일을 JSON으로
        IPmans = faker_generator() # Fake 인간 생성
        data = set_Ipman(IPmans)# Fake리스트를 dic형식으로 변환
        create_json_file(data) #json파일저장
        print('생성완료')
        print('생성완료')

    elif ((act_flag ==1) and (form_flag == 3)):   #Pickle로
        IPmans = faker_generator() # Fake 인간 생성
        save_pickle(IPmans) #피클파일 저장
        print('생성완료')

    elif ((act_flag ==1) and (form_flag == 4)):
        with open('./secure_data.json', 'r', encoding='utf-8') as f:
            secret = json.load(f)

        print(secret)
        connect(secret['host'], secret['dbname'], secret['user'], secret['password'], secret['port'])


    #이전 파일 추가
    elif ((act_flag ==2) and (form_flag == 1)):   #엑셀로
        title=input("추가할 엑셀파일명을 입력해주세요: ")
        IPmans = faker_generator()

        wb = xl.load_workbook('./result/' + title + '.xlsx')
        count = xl.get_Count_Ipmans(wb['data'])
        print('엑셀에 원래 있던 Ipman 수 :', count)
        xl.set_Ipmans(wb['data'], IPmans, start=count+2)
        xl.save_excel(wb)
        print("추가완료")


    elif ((act_flag ==2) and (form_flag == 2)):   #JSON으로
        title=input("추가할 Json파일명을 입력해주세요: ")
        IPmans = faker_generator()

        newContent = set_Ipman(IPmans)  #새로 만들어서 dic으로 만든 데이터 [ dic, dic, dic]
        resContent = []
        with open('./result/' + title + '.json', 'r', encoding='utf-8') as f:
            resContent = json.load(f) #기존에 존재하는 데이터 [ dic, dic, dic]
        content = newContent + resContent
        create_json_file(content) #json파일저장
        print('생성완료')


    elif (act_flag ==2 and form_flag == 3):   #Pickle로
        title=input("추가할 Pickle파일명을 입력해주세요: ")
        newContent = faker_generator() #리스트형식
        with open('./result/'+ title + '.pickle', 'rb') as f:
            resContent = pickle.load(f)  #리스트형식
        content =newContent + resContent
        save_pickle(content) #피클파일 저장
        print('생성완료')


    #파일 확인
    elif ((act_flag ==3) and (form_flag == 1)):   #엑셀로
        title=input("찾는 엑셀파일명을 입력해주세요: ")
        wb = xl.load_workbook('./result/' + title + '.xlsx')
        count = xl.get_Count_Ipmans(wb['data'])
        IPmans = serialization(xl.get_data(wb['data'], count))
        show_whole(IPmans)

    elif (act_flag ==3 and form_flag == 2):   #JSON으로
        title=input("찾는 json파일명을 입력해주세요: ")
        with open('./result/' + title + '.json', 'r', encoding='utf-8') as f:
            data = json.load(f)
        IPmans = serialization(data)
        show_whole(IPmans)

    elif (act_flag ==3 and form_flag == 3):   #Pickle로
        title=input("찾는 피클파일명을 입력해주세요: ")
        with open('./result/'+ title + '.pickle', 'rb') as f:
            data = pickle.load(f)
        show_whole(data)


    else:
        print("재실행 후, 번호를 하나씩만 골라주세요. ")