Skip to content

Instantly share code, notes, and snippets.

@andramarkov
Last active May 4, 2020 10:53
Show Gist options
  • Save andramarkov/0a38e08d6cd5863c80b27e8cffd59869 to your computer and use it in GitHub Desktop.
Save andramarkov/0a38e08d6cd5863c80b27e8cffd59869 to your computer and use it in GitHub Desktop.
Ambil data historis EOD via yahoo, hasil nya dalam bentuk *.csv yang tersimpan di desktop (saya pakai mac), csv ini bisa di import dari amibroker
'''ini prinsipnya sama dengan EOD_direct_to_db_merged.py, bedanya ini nonmerged, pake ini karena eod model lama sering error 'date' '''
import sys
sys.path.append('/Users/andra/Downloads/adrx_repo/bitbucket/newpystock/reorg')
from datetime import datetime, timedelta
import pandas as pd
#agar bisa kayak gini maka harus bikin __init__.py di folder
from lib.yfx_fakeuseragent_modami import IntraData as ambil
import pymysql
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, create_engine, inspect
from pathlib import Path
import datetime
import os
import errno
import time
from random import randint
def newdir(d):
try:
os.makedirs(d)
print ('Processing directory')
print ('Directory created\n' + d)
except OSError as exception:
if exception.errno != errno.EEXIST:
raise
else:
print ("Directory already exists\n%s " % d)
return d
'''periode = str(input("Periode (#opsi: 1d,5d,1mo,3mo,6mo,1y,2y,5y,10y,ytd,max) ? [default: 7d] "))
if len(periode) == 0 :
periode = '7d'
interval = str(input("Interval (#opsi: 1m, 2m, 5m, 15m, 30m, 60m, 90m, 1h, 1d, 5d, 1wk, 1mo, 3mo? [default: 1m] "))
if len(interval) == 0 :
interval = '1m' '''
periode = '1d'
interval = '1d'
#nanti utk hapus .jk dari ticker
def splito(tck):
tck, jk = tck.split('.')
tck = tck.upper()
return tck
myQuery = '''SELECT kode from tblMasterTicker'''
engine = create_engine("mysql+pymysql://{user}:{pw}@localhost/{db}"
.format(user="root",
pw="root",
db="ombandaroo"))
symbols = pd.read_sql_query(myQuery, engine)
my_list = []
for kode in symbols['kode']:
my_list.append(kode)
string = '.JK'
my_new_list = [x + string for x in my_list]
hari_ini = datetime.datetime.today()
sekarang = hari_ini.strftime('%Y-%m-%d')
#desktop
dpath = str(Path.home())
dpath = dpath + '/Desktop/stockdata/eod_all_' + sekarang + '/'
print ('Membuat folder di desktop')
dpath = newdir(dpath)
try:
'''bikin dataframe kosong dulu gan, kalo gak di bikin di awal nanti error kyk tadi
ValueError: If using all scalar values, you must pass an index'''
df = pd.DataFrame()
length = len(my_new_list)
for i in range(length):
try:
#print ('Timestamp: ' + str(datetime.datetime.today()))
lst = my_new_list[i]
nextlst = lst
nextlst = splito(nextlst)
cariapa = ambil(lst, berapa_lama = periode, interval = interval, dropna='True')
tckr = splito(lst) #hapus dot JK dari ticker
df = cariapa.hasil #result dataframe
saveidx = df.index #index bawaan = tanggal
idx = 0
rxx = saveidx.strftime("%Y-%m-%d")
#insert kolom
df.insert(loc=idx, column='Ticker', value=tckr)
df.insert(loc=idx+1, column='Date', value=rxx)
df.insert(loc=idx+7, column='AdjClose', value=df.Close)
#hapus index lama
df.reset_index(drop=True, inplace=True)
print ('Processing ' + tckr.upper() + ' (' + str(i+1) + ' of ' + str(length) + ')')
df.to_csv((os.path.join(dpath,'{}.csv')).format(tckr.upper()), index=False, header=True)
#print ('Sleep@MainLoop')
time.sleep(randint(0,2))
#print ('WakeUp@MainLoop')
except:
print ('Processing ' + nextlst + ' (' + str(i+1) + ' of ' + str(length) + ') - error, skipped ' )
pass #biar tetap lanjut kalo error tickernya
print ('')
print ('Proses selesai')
print ('Directory: ' + dpath)
print ('Timestamp: ' + str(datetime.datetime.today()))
except Exception as e:
print (f'Gagal, Kode Error: {e}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment