astock.py

# -*- coding: utf-8 -*-
import sys, os
import requests
import re
import time
from aclass import *

ResultSuccess = 0
ResultTimeout = 1
ResultNoChange = 2
ResultUnknown = 3

stockList = []
timePattern = re.compile(r',(\d{2}:\d{2}:\d{2}),')
stockPattern = re.compile(r'var hq_str_s[hz]\d{6}="([^,"]+),([^,"]+),([^,"]+),([^,"]+),([^,"]+),([^,"]+),[^,"]+,[^,"]+,[^,"]+,[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,([^,"]+),[^,"]+,[^"]+";')
lastTime = ''
lastData = []

def loadStockList():
	if len(sys.argv) < 2:
		inputFile = open(os.path.join(sys.path[0], 'input.list'), 'r')
		inputList = inputFile.read().split('\n')
		inputFile.close()
	else:
		inputList = []
		for index in range(1, len(sys.argv)):
			inputList.append(sys.argv[index])
	for stockNumber in inputList:
		if len(stockNumber) == 8:
			# 8位长度的代码必须以sh或者sz开头，后面6位是数字
			if (stockNumber.startswith('sh') or stockNumber.startswith('sz')) and stockNumber[2:8].isdecimal():
				stockList.append(stockNumber)
		elif len(stockNumber) == 6:
			# 6位长度的代码必须全是数字
			if stockNumber.isdecimal():
				# 0开头自动补sz，6开头补sh，3开头补sz，否则无效
				if stockNumber.startswith('0'):
					stockList.append('sz' + stockNumber)
				elif stockNumber.startswith('6'):
					stockList.append('sh' + stockNumber)
				elif stockNumber.startswith('3'):
					stockList.append('sz' + stockNumber)
		elif stockNumber == 'sh':
			stockList.append('sh000001')
		elif stockNumber == 'sz':
			stockList.append('sz399001')
		elif stockNumber == 'zx':
			stockList.append('sz399005')
		elif stockNumber == 'cy':
			stockList.append('sz399006')
		elif stockNumber == '300':
			stockList.append('sh000300')
	if len(stockList) == 0:
		return False
	return True

def requestStockData():
	url = 'https://hq.sinajs.cn/list=' + ','.join(stockList)
	try:
		content = requests.get(url, timeout = 3).text
	except requests.exceptions.RequestException:
		return ResultTimeout
	# 判断数据时间有没有更新
	global lastTime
	match = timePattern.search(content)
	if match == None:
		return ResultUnknown
	elif match.group(1) == lastTime:
		return ResultNoChange
	lastTime = match.group(1)
	# 抓取所有数据并计算
	lastData[:] = []
	match = stockPattern.search(content)
	while match:
		stock = Stock(match.group(1), match.group(2), match.group(3), match.group(4), match.group(5), match.group(6))
		stock.calcBuyPercent([match.group(7), match.group(8), match.group(9), match.group(10), match.group(11), match.group(12), match.group(13), match.group(14), match.group(15), match.group(16)])
		lastData.append(stock)
		match = stockPattern.search(content, match.end())
	if len(lastData) == 0:
		return ResultUnknown
	return ResultSuccess

if loadStockList() == False:
	print('没有有效的股票代码')
	print('使用示例: python astock.py sh600000 sz000001')
	print('自动补全: 6字头股票代码脚本会自动补sh前缀，0字头和3字头补sz')
	print('特殊代码: sh-上证指数，sz-深证指数，zx-中小板指，cy-创业板指，300-沪深300')
else:
	while True:
		result = requestStockData()
		if result == ResultSuccess:
			print(lastTime)
			for stock in lastData:
				stock.printStockData()
			time.sleep(20)
		elif result == ResultTimeout:
			print('错误重试')
			time.sleep(5)
		elif result == ResultNoChange:
			timeValue = int(lastTime.replace(':', '')[0:4])
			if (timeValue >= 1130 and timeValue < 1250) or timeValue < 920:
				# 开市前10分钟外
				time.sleep(600)
			elif (timeValue >= 1250 and timeValue < 1300) or timeValue < 930:
				# 开市前十分钟内
				time.sleep(60)
			elif timeValue > 1500:
				print('已经休市')
				break
			else:
				time.sleep(10)
		else:
			print('未知错误')
			break