forked from sljh0214/quantpython
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathex1.py
117 lines (110 loc) · 3.21 KB
/
ex1.py
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
import requests
import bs4
import pandas as pd
import time
from utils_magic import *
#file_path = '마법공식 데이터.xlsx'
#path = 'data.xls'
'''
firmcode_list = ['A005930', 'A005380', 'A035420', 'A003550', 'A034730']
for num, code in enumerate(firmcode_list):
fs_df = make_fs_dataframe(code)
fs_df_changed = change_df(code, fs_df)
if num == 0 :
total_fs = fs_df_changed
else:
total_fs = pd.concat([total_fs, fs_df_changed])
print(total_fs)
'''
'''
firmcode_list = ['A005930', 'A005380', 'A035420', 'A003550', 'A034730']
for num, code in enumerate(firmcode_list):
fr_df = make_fr_dataframe(code)
fr_df_changed = change_df(code, fr_df)
if num == 0 :
total_fr = fr_df_changed
else:
total_fr = pd.concat([total_fr, fr_df_changed])
print(total_fr)
'''
'''
firmcode_list = ['A005930', 'A005380', 'A035420', 'A003550', 'A034730']
for num, code in enumerate(firmcode_list):
invest_df = make_invest_dataframe(code)
invest_df_changed = change_df(code, invest_df)
if num == 0 :
total_invest = invest_df_changed
else:
total_invest = pd.concat([total_invest, invest_df_changed])
print(total_invest)
'''
path = 'data.xls'
code_data = pd.read_excel(path)
code_data = code_data[['종목코드', '기업명']]
code_data['종목코드'] = code_data['종목코드'].apply(make_code)
'''
# 재무제표데이터
for num, code in enumerate(code_data['종목코드']):
try:
print(num, code)
time.sleep(1)
try:
fs_df = make_fs_dataframe(code)
except requests.exceptions.Timeout:
time.sleep(60)
fs_df = make_fs_dataframe(code)
fs_df_changed = change_df(code, fs_df)
if num == 0 :
total_fs = fs_df_changed
else:
total_fs = pd.concat([total_fs, fs_df_changed])
except ValueError:
continue
except KeyError:
continue
total_fs.to_excel('재무제표데이터.xlsx')
'''
'''
# 재무비율데이터
for num, code in enumerate(code_data['종목코드']):
try:
print(num, code)
time.sleep(1)
try:
fr_df = make_fr_dataframe(code)
except requests.exceptions.Timeout:
time.sleep(60)
fr_df = make_fr_dataframe(code)
fr_df_changed = change_df(code, fr_df)
if num == 0 :
total_fr = fr_df_changed
else:
total_fr = pd.concat([total_fr, fr_df_changed])
except ValueError:
continue
except KeyError:
continue
total_fr.to_excel('재무비율데이터.xlsx')
'''
'''
# 투자지표데이터
for num, code in enumerate(code_data['종목코드']):
try:
print(num, code)
time.sleep(1)
try:
invest_df = make_invest_dataframe(code)
except requests.exceptions.Timeout:
time.sleep(60)
invest_df = make_invest_dataframe(code)
invest_df_changed = change_df(code, invest_df)
if num == 0 :
total_invest = invest_df_changed
else:
total_invest = pd.concat([total_invest, invest_df_changed])
except ValueError:
continue
except KeyError:
continue
total_invest.to_excel('투자지표데이터.xlsx')
'''