Adds data dir config

otto/config.py

@@ -1,7 +1,7 @@
 from os import environ
 
 LOG_DIR = environ.get('LOG_DIR', 'logs')
-
+DATA_DIR = environ.get('DATA_DIR', 'data')
 
 defaults = {
  'fontcolor': '#FFFFFF',

otto/getdata.py

@@ -1,11 +1,12 @@
-from csv import reader
 from hashlib import sha256
 from json import loads
 from os import path
 from time import strftime
 
 from requests import head
 
+from otto.config import DATA_DIR
+from otto.log import log
 from otto.utils import download_url
 
 content_types = ['image/jpeg', 'video/mp4', 'image/png', 'audio/mpeg']
@@ -16,7 +17,7 @@ def sha(s):
  return sha256(s.encode()).hexdigest()
 
 
-def download(url, location='data'):
+def download(url, location=DATA_DIR):
  if not url.startswith('http'):
  return url
  h = head(url, allow_redirects=True)
@@ -45,6 +46,8 @@ def download(url, location='data'):
 
 
 def openCsv(path):
+ from csv import reader
+
  csvreader = reader(open(path, 'r'))
  d = {}
  for row in csvreader:
@@ -75,5 +78,7 @@ def urlToJson(path):
  source = response.read()
  data = loads(source)
  else:
- print('An error occurred while attempting to retrieve data from the API.')
+ log.error(
+ 'An error occurred while attempting to retrieve data from the API.'
+ )
  return data

otto/preview.py

@@ -3,6 +3,7 @@
 from otto.render import generateEdl
 from otto.getdata import timestr
 from os.path import join
+from otto.config import DATA_DIR
 
 previewAPI = APIRouter()
 
@@ -22,7 +23,7 @@ async def previewFrame(t: float, render: Render):
  video = generateEdl(
  Edl(clips=active_clips), moviesize=(render.width, render.height)
  )
- frame_name = join('data', timestr() + '.jpg')
+ frame_name = join(DATA_DIR, timestr() + '.jpg')
  print(f'saving frame {t} as: {frame_name}', video)
  video.save_frame(frame_name, t=t, withmask=False)
  return frame_name