From 1849f75ad04a8a50c73862a6d0794fe65016343c Mon Sep 17 00:00:00 2001
From: Harry Marr <harry.marr@gmail.com>
Date: Tue, 31 Aug 2010 00:23:59 +0100
Subject: [PATCH] Made GridFSProxy a bit stricter / safer

---
 mongoengine/fields.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/mongoengine/fields.py b/mongoengine/fields.py
index ffcfb53d..418f57cc 100644
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -511,6 +511,10 @@ class BinaryField(BaseField):
             raise ValidationError('Binary value is too long')
 
 
+class GridFSError(Exception):
+    pass
+
+
 class GridFSProxy(object):
     """Proxy object to handle writing and reading of files to and from GridFS
     """
@@ -541,12 +545,18 @@ class GridFSProxy(object):
         self.grid_id = self.newfile._id
 
     def put(self, file, **kwargs):
+        if self.grid_id:
+            raise GridFSError('This document alreay has a file. Either delete '
+                              'it or call replace to overwrite it')
         self.grid_id = self.fs.put(file, **kwargs)
 
     def write(self, string):
-        if not self.newfile:
+        if self.grid_id:
+            if not self.newfile:
+                raise GridFSError('This document alreay has a file. Either '
+                                  'delete it or call replace to overwrite it')
+        else:
             self.new_file()
-            self.grid_id = self.newfile._id
         self.newfile.write(string)
 
     def writelines(self, lines):