finish hollow_and_filled tutorial; add mpf.write_style_file()

Author: DanielGoldfarb

examples/hollow_and_filled_candles.ipynb

@@ -62,3 +62,33 @@
 mpf.plot(df ,style=st3,**kwargs,savefig='hollow_black_white.jpg')
 mpf.plot(dfs,style=st3,**kwargs,savefig='solid_black_white.jpg')
 
+##  
+##  df = pd.read_csv('data/SP500_NOV2019_Hist.csv',index_col=0,parse_dates=True)
+##  
+##  df.loc['2019-11-07','Open' ] = 3085
+##  df.loc['2019-11-07','Close'] = 3092
+##  df.loc['2019-11-07','Low'  ] = 3081
+##  df.loc['2019-11-07','High' ] = 3096
+##  
+##  df.loc['2019-11-08','Open' ] = 3085
+##  df.loc['2019-11-08','Close'] = 3092
+##  df.loc['2019-11-08','Low'  ] = 3081
+##  df.loc['2019-11-08','High' ] = 3096
+##  
+##  
+##  dfs = df.copy()
+##  dfs.loc['2019-11-07','Open' ] = 3092
+##  dfs.loc['2019-11-07','Close'] = 3085
+##  
+##  dfs.loc['2019-11-08','Open' ] = 3092
+##  dfs.loc['2019-11-08','Close'] = 3085
+##  
+##  neoclassic = mpf.make_mpf_style(base_mpf_style='classic',facecolor='cyan') 
+##  st = neoclassic
+##  st = mpf.make_mpf_style(base_mpf_style='nightclouds',gridstyle='')
+##  #st = mpf.make_mpf_style(base_mpf_style='sas',gridstyle='') #figcolor='#3C8284'
+##  st1 = mpf.make_mpf_style(base_mpf_style='classic',gridstyle='',facecolor='#56b0b3')
+##  #st1 = mpf.make_mpf_style(base_mpf_style='classic',gridstyle='',facecolor='#ffd6dd')
+##  mpf.plot(df.iloc[3:8],type='hollow_candle',volume=False,style=st1,figscale=1,update_width_config=dict(candle_linewidth=2.25))
+##  mpf.plot(dfs.iloc[3:8],type='hollow_candle',volume=False,style=st1,figscale=1,update_width_config=dict(candle_linewidth=2.25))
+##  

examples/scratch_pad/hollow_and_filled_images.py

@@ -1,5 +1,6 @@
 import mplfinance._mpf_warnings
 from   mplfinance.plotting  import plot, make_addplot
-from   mplfinance._styles   import make_mpf_style, make_marketcolors, available_styles
+from   mplfinance._styles   import make_mpf_style, make_marketcolors
+from   mplfinance._styles   import available_styles, write_style_file
 from   mplfinance._version  import __version__
 from   mplfinance._mplwraps import figure, show

src/mplfinance/__init__.py

@@ -1,6 +1,8 @@
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
 import copy
+import pprint
+import os.path as path
 
 from   mplfinance._arg_validators import _process_kwargs, _validate_vkwargs_dict
 from   mplfinance._styledata      import _styles
@@ -271,3 +273,34 @@ def _check_and_set_mktcolor(candle,**kwarg):
         marketcolors.update({'alpha':config['alpha']})
 
     return marketcolors
+
+def write_style_file(style,filename):
+    pp   = pprint.PrettyPrinter(indent=4,sort_dicts=False,compact=True)
+    strl = pp.pformat(style).splitlines()
+
+    if not isinstance(style,dict):
+        raise TypeError('Specified style must be in `dict` format')
+
+    if path.exists(filename):
+        print('"'+filename+'" exists.') 
+        answer = input(' Overwrite(Y/N)? ')
+        a = answer.lower()
+        if a != 'y' and a != 'yes':
+            raise FileExistsError
+
+    f = open(filename,'w')
+    f.write('style = '+strl[0].replace('{','dict(',1).replace("'","",2).replace(':',' =',1)+'\n')
+    for line in strl[1:-1]:
+        if "'" in line[0:5]:
+            f.write('            '+line.replace("'","",2).replace(':',' =',1)+'\n')
+        else:
+            f.write('            '+line+'\n')
+    line = strl[-1]
+    if "'" in line[0:5]:
+        line = line.replace("'","",2).replace(':',' =',1)[::-1]
+    else:
+        line = line[::-1]
+    f.write('            '+line.replace('}',')',1)[::-1]+'\n')
+    f.close()
+    print('Wrote style file "'+filename+'"')
+    return