mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-29 20:33:46 +08:00
Joe Jevnik
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
#
|
|
# Copyright 2015 Quantopian, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import os
|
|
|
|
|
|
if os.name == 'nt':
|
|
# we need to be able to write to our temp directoy on windows so we
|
|
# create a subdir in %TMP% that has write access and use that as %TMP%
|
|
def _():
|
|
import atexit
|
|
import tempfile
|
|
|
|
tempfile.tempdir = tempdir = tempfile.mkdtemp()
|
|
|
|
@atexit.register
|
|
def cleanup_tempdir():
|
|
import shutil
|
|
shutil.rmtree(tempdir)
|
|
_()
|
|
del _
|
|
|
|
|
|
# This is *not* a place to dump arbitrary classes/modules for convenience,
|
|
# it is a place to expose the public interfaces.
|
|
from . import data
|
|
from . import finance
|
|
from . import gens
|
|
from . import utils
|
|
from .utils.run_algo import run_algorithm
|
|
from ._version import get_versions
|
|
# These need to happen after the other imports.
|
|
from . algorithm import TradingAlgorithm
|
|
from . import api
|
|
|
|
__version__ = get_versions()['version']
|
|
del get_versions
|
|
|
|
|
|
def load_ipython_extension(ipython):
|
|
from .__main__ import zipline_magic
|
|
ipython.register_magic_function(zipline_magic, 'line_cell', 'zipline')
|
|
|
|
|
|
__all__ = [
|
|
'TradingAlgorithm',
|
|
'api',
|
|
'data',
|
|
'finance',
|
|
'gens',
|
|
'run_algorithm',
|
|
'utils',
|
|
]
|