def event_init_interest():
from venus.stock_interest import EventInterest
from venus.stock_base2 import resolve_stock_list
stock_list = resolve_stock_list('stock')
event = EventInterest(GLOBAL_HEADER)
event._load_template()
for stock_code in stock_list:
event.record_interest(stock_code)
delay(10)
def event_record_company_stock_structure():
event = EventCompany(GLOBAL_HEADER)
stock_list = event.get_all_stock_list()
for stock in stock_list:
try:
# print(stock)
event.record_stock_structure(stock)
delay(10)
except Exception as e:
ERROR(e)
def event_init_stock():
"""
Init database from a blank stock list.
"""
from venus.stock_manager2 import EventTradeDataManager
from venus.stock_base2 import resolve_stock_list
stock_list = resolve_stock_list('totalstocklist')
event = EventTradeDataManager(GLOBAL_HEADER)
for stock_code in stock_list:
event.init_stock_data(stock_code)
delay(5)
def event_download_news(n):
from mars.network import delay
from polaris.mysql8 import mysqlHeader
from taurus.news_downloader import newsSpider
header = mysqlHeader('stock', 'stock2020', 'natural_language')
event = newsSpider(header)
url_list = event.get_url_list()
for url in url_list[:n]:
try:
event.save_page(url)
delay(5)
except Exception as e:
print(url)
print(e)
def event_download_detail_data(trade_date_list: list = None):
import datetime
from jupiter.network import delay
from venus.stock_base2 import resolve_stock_list
from venus.stock_manager2 import EventTradeDataManager
stock_list = resolve_stock_list('stock')
event = EventTradeDataManager(GLOBAL_HEADER)
today = datetime.date.today()
if not trade_date_list:
trade_date_list = [
(today - datetime.timedelta(days=i)).strftime('%Y%m%d')
for i in range(1, 6)
]
stock_list = event.get_all_stock_list()
for trade_date in trade_date_list:
for stock in stock_list:
# print(f"Download detail trade data {stock}: {trade_date}")
event.get_trade_detail_data(stock, trade_date)
delay(3)
def unit_test_network():
network.delay(4)
print('Network test finished.')