import os.path
from django.conf import settings
from pymongo import Connection
from gridfs import GridFS
import pyfits


def fetch(filename):
    mongo_connection = Connection(host=settings.MONGODB["host"],
                                  port=settings.MONGODB["port"])
    gfs = GridFS(mongo_connection[settings.MONGODB["database"]])
    return gfs.get_version(filename)


def get_hdu(url):
    if settings.MONGODB["enabled"]:
        mongo_file = tkpdb.mongo.fetch(url)
        return pyfits.open(mongo_file, mode="readonly")
    elif os.path.exists(url):
        return pyfits.open(url, readonly=True)
    else:
        raise Exception("Can't find file")
import os.path
from django.conf import settings
from pymongo import Connection
from gridfs import GridFS
import pyfits


def fetch(filename):
    mongo_connection = Connection(host=settings.MONGODB["host"],
                                  port=settings.MONGODB["port"])
    gfs = GridFS(mongo_connection[settings.MONGODB["database"]])
    return gfs.get_version(filename)


def get_hdu(url):
    if settings.MONGODB["enabled"]:
        mongo_file = fetch(url)
        return pyfits.open(mongo_file, mode="readonly")
    elif os.path.exists(url):
        return pyfits.open(url, readonly=True)
    else:
        raise Exception("Can't find file")
fix fetch from mongo