MYFI_back/make_data.py at master · Nahaeran/MYFI_back · 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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
# make_data.py 파일은 랜덤한 더미 데이터를 만드는 예시 파일입니다.
# 반드시, 사용하는 필드를 확인한 후 본인의 프로젝트에 맞게 수정하여 진행해야 합니다.

# [참고] 현재 코드는 아래 User 모델을 기준으로 작성되어 있습니다.
# class User(AbstractBaseUser):
#     username = models.CharField(max_length=30, unique=True)
#     nickname = models.CharField(max_length=255, blank=True, null=True)
#     email = models.EmailField(max_length=254, blank=True, null=True)
#     age = models.IntegerField(blank=True, null=True)
#     money = models.IntegerField(blank=True, null=True)
#     salary = models.IntegerField(blank=True, null=True)
#     # 가입한 상품 목록 리스트를 ,로 구분된 문자열로 저장함
#     financial_products = models.TextField(blank=True, null=True)

#     # superuser fields
#     is_active = models.BooleanField(default=True)
#     is_staff = models.BooleanField(default=False)
#     is_superuser = models.BooleanField(default=False)

# import os
# os.environ.setdefault("DJANGO_SETTINGS_MODULE", "fianl_pjt_back.settings")

# import django
# django.setup()

import random
import requests
# from financial_instruments.models import Deposit, Saving
# from financial_instruments.serializers import DepositSerializer, SavingSerializer

first_name_samples = "송나김이박최정강조윤장임공손전한오서신권황안홍"
middle_name_samples = "찬해지수근민서예지도하주윤채현지혜동희의"
last_name_samples = "의란민혜용준윤우원호후서연아은진"

# Deposits = Deposit.objects.all()
# Savings = Saving.objects.all()

def random_name():
    result = ""
    result += random.choice(first_name_samples)
    result += random.choice(middle_name_samples)
    result += random.choice(last_name_samples)
    return result #+ str(random.randint(1, 100))

# 현재 API 에 들어있는 금융 상품 코드 리스트 저장
DP_URL = 'http://finlife.fss.or.kr/finlifeapi/depositProductsSearch.json'
SP_URL = 'http://finlife.fss.or.kr/finlifeapi/savingProductsSearch.json'

API_KEY = '16dd1959199940770f9f4793f549c2a0'

contract_deposit = []
contract_saving = []

params = {
  'auth': API_KEY,
  # 금융회사 코드 020000(은행), 030200(여신전문), 030300(저축은행), 050000(보험), 060000(금융투자)
  'topFinGrpNo': '020000',
  'pageNo': 1
}

# 정기예금 목록 저장
response = requests.get(DP_URL, params=params).json()
baseList = response.get('result').get('baseList')   # 상품 목록

for product in baseList:
    contract_deposit.append(product['fin_prdt_cd'])

# 적금 목록 저장
response = requests.get(SP_URL, params=params).json()
baseList = response.get('result').get('baseList')   # 상품 목록

for product in baseList:
    contract_saving.append(product['fin_prdt_cd'])

dict_keys = ['username', 'gender', 'contract_deposit', 'contract_saving', 'age', 'money', 'salary']

# json 파일 만들기
import json
from collections import OrderedDict

file = OrderedDict()

name_list = []
N = 5000
i = 0

while i < N:
    rn = random_name()
    if rn in name_list:
        continue

    name_list.append(rn)
    i += 1


# 저장 위치는 프로젝트 구조에 맞게 수정합니다.
save_dir = 'accounts/fixtures/accounts/user_data.json'
with open(save_dir, 'w', encoding="utf-8") as f:

    f.write('[')
    for i in range(N):
        # 랜덤한 데이터를 삽입
        file["model"] = "accounts.User"
        file["pk"] = i+1
        file["fields"] = {
            'username': 'test'+str(i), # 유저아이디 test1,2,3,4,5,6,7
            'name': name_list[i],  # 유저 이름 랜덤 생성
            # 랜덤한 0~5개의 상품을 가입하도록 삽입됨

            # 'contract_deposit': [ DepositSerializer(random.choice(Deposits)).data for _ in range(random.randint(1, 2))], # 금융 상품 리스트
            # 'contract_saving': [ SavingSerializer(random.choice(Savings)).data for _ in range(random.randint(1, 3))], # 금융 상품 리스트

            'contract_deposit': list(set([ random.randrange(1,38) for _ in range(2) ])) , # 금융 상품 리스트
            'contract_saving': list(set([ random.randrange(1,62) for _ in range(random.randint(2, 3)) ])), # 금융 상품 리스트
            # 'contract_deposit': '[' + ','.join([str(random.randrange(1,38)) for _ in range(random.randint(1, 2))]) + ']', # 금융 상품 리스트
            # 'contract_saving': '[' +','.join([str(random.randrange(1,62)) for _ in range(random.randint(1, 3))]) + ']', # 금융 상품 리스트
            'age': random.randint(20, 70),  # 나이
            'money': random.randrange(1000000, 100000000, 1000000),    # 현재 가진 금액
            'salary': random.randrange(1000000, 150000000, 12000000), # 연봉
            'password': "a123456789!",
            'desire_amount_saving': random.randrange(50000, 3000000, 100000), #적금 랜덤
            'desire_amount_deposit': random.randrange(100000, 50000000, 1000000), # 예금 랜덤
            'deposit_period':random.choice([6, 12, 24, 36]), # 기간 랜덤
            'saving_period':random.choice([6, 12, 24, 36]), # 기간 랜덤
            'is_active': True,
            'is_staff': False,
            'is_superuser': False
        }
        file['fields']['desire_amount_deposit'] = random.randrange(100000, file['fields']['money'], 1000000)

        json.dump(file, f, ensure_ascii=False, indent="\t")
        if i != N-1:
            f.write(',')

    f.write(']')
    f.close()

print(f'데이터 생성 완료 / 저장 위치: {save_dir}')