<filename>ally/__init__.py<gh_stars>0
#################################################
"""			ALLY				"""
#################################################

import pyximport; pyximport.install()
from . import 	order as order_utils
from . import 	option_info
from . import 	watchlist as watchlist_utils
from . import	instrument
from . import 	api_calls
from . import	utils
from . import 	fixml
from . import	news

all = ['fixml.FIXML', 'order', 'instrument', 'Ally', 'utils']

from datetime import timedelta
from json import loads
from os import environ


############################################################################
class Ally:
	endpoints={
		'base'					:'https://api.tradeking.com/v1/',
		'request_token'			:'https://developers.tradeking.com/oauth/request_token',
		'user_auth'				:'https://developers.tradeking.com/oauth/authorize',
		'resource_owner_key'	:'https://developers.tradeking.com/oauth/resource_owner_key',
		'stream'				:'https://stream.tradeking.com/v1/'
	}
	json_params = {
		'indent':4
	}
	
	# Cache Oauth requests (faster)
	last_auth_time = None
	auth		   = None
	valid_auth_dt  = timedelta(seconds=9.7)

	############################################################################
	# Option calls are stored in option_info.py
	get_strike_prices	=	option_info.get_strike_prices
	get_exp_dates		=	option_info.get_exp_dates
	search_options		=	option_info.search_options
	options_chain		=	option_info.options_chain
	############################################################################
	# Most API calls are stored in api_calls.py
	# Account operations
	get_accounts		=	api_calls.get_accounts
	get_holdings		=	api_calls.get_holdings
	holdings_chart		=	api_calls.holdings_chart
	submit_order		=	api_calls.submit_order
	account_history		=	api_calls.account_history
	order_history		=	api_calls.order_history
	get_cash			=	api_calls.get_cash
	############################################################################
	# Pricing information
	get_quote			=	api_calls.get_quote
	timesales			=	api_calls.timesales
	quote_stream		=	api_calls.quote_stream # Unfinished
	############################################################################
	# Common
	create_auth			=	api_calls.create_auth
	call_api			=	api_calls.call_api
	req_sess			=	api_calls.req_sess
	############################################################################
	# Utilities
	market_clock		=	api_calls.market_clock
	api_status			=	api_calls.api_status
	get_member			=	api_calls.get_member
	toplists			=	api_calls.toplists
	############################################################################
	# Watchlists
	get_watchlists		=	watchlist_utils.get_watchlists
	new_watchlist		=	watchlist_utils.new_watchlist
	delete_watchlist	=	watchlist_utils.delete_watchlist
	watchlist			=	watchlist_utils.watchlist
	delete_symbol		=	watchlist_utils.delete_symbol
	add_symbol			=	watchlist_utils.add_symbol
	############################################################################
	# News
	news_search			=	news.news_search
	############################################################################
	def __init__(self, params=None ):
		self.holdings_graph	= None
		self.holdings		= None
		self.accounts		= []
		self.session		= None
		
		
		try:
			
			# We were passed a JSON file
			if type(params) == type(""):
				with open(params, 'r') as f:
					params = load(f)
					
			# SET paramS
			if type(params) == type({}):
				self.params = params

			else:
				# Try to use environment params
				self.params = {
					'resource_owner_secret'	:	environ['ALLY_OAUTH_SECRET'],
					'resource_owner_key'	:	environ['ALLY_OAUTH_TOKEN'],
					'client_secret'			:	environ['ALLY_CONSUMER_SECRET'],
					'client_key'			:	environ['ALLY_CONSUMER_KEY'],
				}
				
				if 'ALLY_ACCOUNT_NBR' in environ:
					self.params['account'] = environ['ALLY_ACCOUNT_NBR']

			## Check that we have all the parameters we need
			for t in ( 
				'resource_owner_secret',
				'resource_owner_key',
				'client_secret',
				'client_key'
			):
				if t not in self.params.keys():
					raise
				
		except:
			print(
				"""Didn't specify parameters or environment variables not set!
				Go to https://github.com/alienbrett/PyAlly.git for help"""
			)
			raise Exception("Didn't specify Ally API environment varialbles!")
			
