Expose plotting logic in a local Python module

candlestick_chart_example.py

@@ -1,8 +1,6 @@
 from database.ohlc import ohlc
 from datetime import datetime, timedelta
-
+from plotting import CandlestickChart, plot
-from candlestick_chart import CandlestickChart
-from plot import plot
 
 end_date = datetime.today().date()
 start_date = (end_date - timedelta(days = 90))

line_chart_example.py

@@ -1,9 +1,6 @@
 from database.ohlc import ohlc
 from datetime import datetime, timedelta
-
+from plotting import Line, LineChart, plot
-from line import Line
-from line_chart import LineChart
-from plot import plot
 
 end_date = datetime.today().date()
 start_date = (end_date - timedelta(days = 90))

multi_line_chart_example.py

@@ -1,9 +1,6 @@
 from database.ohlc import ohlc
 from datetime import datetime, timedelta
-
+from plotting import Line, LineChart, plot
-from line import Line
-from line_chart import LineChart
-from plot import plot
 
 end_date = datetime.today().date()
 start_date = (end_date - timedelta(days = 90))

plotting/__init__.py

@@ -0,0 +1,5 @@
+from .backtest_chart import BacktestChart
+from .candlestick_chart import CandlestickChart
+from .line import Line
+from .line_chart import LineChart
+from .plot import figure_with_subplots, plot

plotting/backtest_chart.py

@@ -1,9 +1,10 @@
-from chart import Chart
 from numpy import where
 from pandas import Series
 from plotly.graph_objects import Bar
 from typing import List
 
+from .chart import Chart
+
 class BacktestChart(Chart):
 
     def __init__(

plotting/candlestick_chart.py

@@ -1,8 +1,9 @@
-from chart import Chart
 from pandas import Series
 from plotly.graph_objects import Candlestick
 from typing import List
 
+from .chart import Chart
+
 class CandlestickChart(Chart):
 
     def __init__(

plotting/line_chart.py

@@ -1,9 +1,10 @@
-from chart import Chart
-from line import Line
 from pandas import Series
 from plotly.graph_objects import Scatter
 from typing import List
 
+from .chart import Chart
+from .line import Line
+
 class LineChart(Chart):
 
     def __init__(self, x: Series, lines: List[Line], title: str = ''):

plotting/plot.py

@@ -1,8 +1,9 @@
-from chart import Chart
 from plotly.graph_objects import Figure
 from plotly.subplots import make_subplots
 from typing import List
 
+from .chart import Chart
+
 def subplot_titles(subplots: List[List[Chart]]) -> List[str]:
     subplot_titles = []
     for row in subplots:

setup.py

@@ -0,0 +1,12 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="plotting",
+    version="1.0",
+    packages=find_packages(),
+    install_requires=[
+        'numpy',
+        'pandas',
+        'plotly'
+    ],
+)

subplot_example.py

@@ -1,10 +1,6 @@
 from database.ohlc import ohlc
 from datetime import datetime, timedelta
-
+from plotting import CandlestickChart, figure_with_subplots, Line, LineChart
-from candlestick_chart import CandlestickChart
-from line import Line
-from line_chart import LineChart
-from plot import figure_with_subplots
 
 end_date = datetime.today().date()
 start_date = (end_date - timedelta(days = 90))