Move storage instatiation

gedgo/gedcom_update.py

@@ -12,15 +12,12 @@
 
 from PIL import Image
 
-gedcom_storage = None
+from gedgo.storages import gedcom_storage
 
 
 @transaction.atomic
 def update(g, file_name, verbose=True):
     # Prevent circular dependencies
-    global gedcom_storage
-    from gedgo.views.util import gedcom_storage as gs
-    gedcom_storage = gs
     if verbose:
         print 'Parsing content'
     parsed = GedcomParser(file_name)

gedgo/storages.py

@@ -1,6 +1,7 @@
 from django.core.files.storage import Storage
 from django.utils._os import safe_join
 from django.conf import settings
+from django.utils.module_loading import import_string
 
 import os
 from dropbox.client import DropboxClient
@@ -54,3 +55,10 @@ def size(self, name):
 
     def url(self, name):
         return self.client.media(self.path(name))['url']
+
+
+research_storage = import_string(settings.GEDGO_RESEARCH_FILE_STORAGE)(
+    location=settings.GEDGO_RESEARCH_FILE_ROOT)
+
+gedcom_storage = import_string(settings.GEDGO_GEDCOM_FILE_STORAGE)(
+    location=settings.GEDGO_GEDCOM_FILE_ROOT)

gedgo/views/research.py

@@ -4,7 +4,8 @@
 from os import path
 import mimetypes
 
-from gedgo.views.util import serve_content, render, research_storage as storage
+from gedgo.views.util import serve_content, render
+from gedgo.storages import research_storage as storage
 
 
 @login_required

gedgo/views/util.py

@@ -9,24 +9,14 @@
 from django.contrib import messages
 from django.shortcuts import render_to_response
 from django.template import RequestContext
-from django.utils.module_loading import import_string
 
 from gedgo.models import BlogPost, Documentary
 from gedgo.forms import CommentForm
+from gedgo.storages import gedcom_storage, research_storage
 
 from os import path
 import mimetypes
 
-research_storage = None
-if getattr(settings, 'GEDGO_RESEARCH_FILE_STORAGE', None):
-    research_storage = import_string(settings.GEDGO_RESEARCH_FILE_STORAGE)(
-        location=settings.GEDGO_RESEARCH_FILE_ROOT)
-
-gedcom_storage = None
-if getattr(settings, 'GEDGO_GEDCOM_FILE_STORAGE', None):
-    gedcom_storage = import_string(settings.GEDGO_GEDCOM_FILE_STORAGE)(
-        location=settings.GEDGO_GEDCOM_FILE_ROOT)
-
 
 STORAGES = {
     'research': research_storage,

settings.py

@@ -29,7 +29,7 @@
 USE_L10N = True
 USE_TZ = True
 
-MEDIA_ROOT = '/src/files/'
+MEDIA_ROOT = '/src/files/default/'
 MEDIA_URL = '/gedgo/media/'
 
 STATIC_ROOT = ''
@@ -102,6 +102,11 @@
 GEDGO_SENDFILE_HEADER = None
 GEDGO_SITE_TITLE = 'My Genealogy Site'
 GEDGO_REDIS_SERVER = 'redis'
+GEDGO_RESEARCH_FILE_STORAGE = 'django.core.files.storage.FileSystemStorage'
+GEDGO_RESEARCH_FILE_ROOT = '/src/files/gedcom/'
+GEDGO_GEDCOM_FILE_STORAGE = 'django.core.files.storage.FileSystemStorage'
+GEDGO_GEDCOM_FILE_ROOT = '/src/files/research/'
+GEDGO_SHOW_RESEARCH_FILES = True
 
 BROKER_BACKEND = 'redis'
 BROKER_URL = 'redis://redis:6379/0'