X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_common%2Ffsmongo.py;h=740d54032c9237986a989c7753820088b36db75b;hb=refs%2Fchanges%2F70%2F9370%2F1;hp=07d4821e942e0df0eb85bbbfb01e4a98b47e5043;hpb=b07e4ef6e44bae6044be7420d982406ab0122275;p=osm%2Fcommon.git diff --git a/osm_common/fsmongo.py b/osm_common/fsmongo.py index 07d4821..740d540 100644 --- a/osm_common/fsmongo.py +++ b/osm_common/fsmongo.py @@ -16,13 +16,16 @@ # contact: eduardo.sousa@canonical.com ## +import errno +from http import HTTPStatus from io import BytesIO, StringIO -from pymongo import MongoClient -from gridfs import GridFSBucket, errors import logging -from http import HTTPStatus import os + +from gridfs import GridFSBucket, errors from osm_common.fsbase import FsBase, FsException +from pymongo import MongoClient + __author__ = "Eduardo Sousa " @@ -215,6 +218,13 @@ class FsMongo(FsBase): self.fs.download_to_stream(writing_file._id, b) b.seek(0) link = b.read().decode("utf-8") + + try: + os.remove(file_path) + except OSError as e: + if e.errno != errno.ENOENT: + # This is probably permission denied or worse + raise os.symlink(link, file_path) else: with open(file_path, 'wb+') as file_stream: @@ -317,7 +327,7 @@ class FsMongo(FsBase): if requested_file.metadata["type"] == mode: return True - + if requested_file.metadata["type"] == "sym" and mode == "file": return True @@ -452,7 +462,7 @@ class FsMongo(FsBase): else: self.fs.delete(requested_file._id) if not found and not ignore_non_exist: - raise FsException("File {} does not exist".format(storage), http_code=HTTPStatus.NOT_FOUND) + raise FsException("File {} does not exist".format(storage), http_code=HTTPStatus.NOT_FOUND) except IOError as e: raise FsException("File {} cannot be deleted: {}".format(f, e), http_code=HTTPStatus.INTERNAL_SERVER_ERROR)