Ubuntu

2

1 import FinanceDataReader as fdr 1 import FinanceDataReader as fdr
2 import pandas as pd 2 import pandas as pd
3 import sys 3 import sys
4 +from fuzzywuzzy import process
5 +
4 # -*- coding: utf-8 -*- 6 # -*- coding: utf-8 -*-
5 7
8 +def get_matches(query, choices, limit=3):
9 + result = process.extract(query, choices, limit=limit)
10 + return result
11 +
6 def basicinform(input): 12 def basicinform(input):
7 stocks = pd.read_csv('stockcodename.csv', index_col=0) 13 stocks = pd.read_csv('stockcodename.csv', index_col=0)
8 symbol = '' 14 symbol = ''
15 +
9 for i in enumerate(stocks.Name): 16 for i in enumerate(stocks.Name):
10 if i[1] == input: 17 if i[1] == input:
11 symbol = (stocks.iloc[i[0]].Symbol) 18 symbol = (stocks.iloc[i[0]].Symbol)
12 break 19 break
13 20
21 + if(symbol == ''):
22 + fuzzy = get_matches(input, stocks.Name)
23 + cand = ''
24 + for i in fuzzy:
25 + cand += i[0]
26 + cand += " "
27 + cand += "중 찾는게 있으신가요? \n다시 입력해주세요."
28 + return cand
29 +
14 df = fdr.DataReader(symbol) 30 df = fdr.DataReader(symbol)
15 ror_df = df.Close.pct_change() 31 ror_df = df.Close.pct_change()
16 volume = df.Volume.iloc[-1] 32 volume = df.Volume.iloc[-1]
...@@ -25,7 +41,5 @@ def basicinform(input): ...@@ -25,7 +41,5 @@ def basicinform(input):
25 return value 41 return value
26 42
27 43
28 -#print(basicinform('삼성전자'))
29 -
30 args = sys.argv 44 args = sys.argv
31 -print(basicinform(args[1])) 45 +print(basicinform(args[0]))
......