Performance boostWe now use json instead of yaml.

.env

@@ -0,0 +1 @@
+PYTHONPATH=src

.gitignore

@@ -18,3 +18,4 @@ report.xml
 build/*
 .pytest_cache/*
 .vscode/*
+src/pytickersymbols/data/stocks.json

requirements-dev.txt

@@ -2,4 +2,5 @@ wheel==0.33.4
 pytest
 pytest-cov
 pandas==0.24.2
-yfinance==0.1.44
+yfinance==0.1.44
+PyYAML==5.1.1

setup.py

@@ -5,17 +5,16 @@
   Use of this source code is governed by an MIT-style license that
   can be found in the LICENSE file.
 """
-
 from setuptools import setup, find_packages
 
 EXCLUDE_FROM_PACKAGES = ['test', 'test.*', 'test*']
-VERSION = '1.1.3'
+VERSION = '1.1.4'
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 INSTALL_REQUIRES = (
-    ['PyYAML==5.1.1', 'wheel==0.33.4']
+    ['wheel==0.33.4']
 )
 
 setup(
@@ -32,7 +31,7 @@
     url="https://github.com/portfolioplus/pytickersymbols",
     packages=find_packages('src', exclude=EXCLUDE_FROM_PACKAGES),
     install_requires=INSTALL_REQUIRES,
-    package_data={'': ['data/*.yaml']},
+    package_data={'': ['data/*.json']},
     include_package_data=True,
     classifiers=[
         'Development Status :: 5 - Production/Stable',

src/pytickersymbols/__init__.py

@@ -6,18 +6,18 @@
   can be found in the LICENSE file.
 """
 import os
-import yaml
+import json
 
 
 class PyTickerSymbols:
 
     def __init__(self):
         self.__stocks = None
-        yaml_path = os.path.join(
-            os.path.dirname(os.path.realpath(__file__)), "data", "stocks.yaml"
+        json_path = os.path.join(
+            os.path.dirname(os.path.realpath(__file__)), "data", "stocks.json"
         )
-        with open(yaml_path) as stocks:
-            self.__stocks = yaml.safe_load(stocks)
+        with open(json_path) as stocks:
+            self.__stocks = json.load(stocks)
 
     def get_all_indices(self):
         """

tox.ini

@@ -7,12 +7,12 @@ envlist = py36,
 [testenv]
 setenv =
     PYTHONPATH = {toxinidir}/src
-deps = -r{toxinidir}/requirements.txt
-       -r{toxinidir}/requirements-dev.txt
+deps = -r{toxinidir}/requirements-dev.txt
 
 basepython =
     py36: python3.6
 commands =
+    python yaml2json.py
     pytest tests/ --cov src/pytickersymbols --cov-report term-missing
     python setup.py bdist_wheel
 

yaml2json.py

@@ -0,0 +1,15 @@
+import yaml
+import json
+import os
+
+# convert yaml file to json
+input_path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
+                          'stocks.yaml')
+output_path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
+                           'src', 'pytickersymbols', 'data', 'stocks.json')
+
+os.makedirs(os.path.dirname(os.path.realpath(output_path)))
+
+with open(output_path, 'w') as out_file:
+    with open(input_path, 'r') as in_file:
+        out_file.write(json.dumps(yaml.safe_load(in_file)))