From 42728d5d676f64b00f6c2ae0b2aa8e131c1448a9 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Sat, 17 Feb 2018 15:07:26 -0800
Subject: [PATCH] Add method Photo.set_override_filename.

---
 etiquette/objects.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/etiquette/objects.py b/etiquette/objects.py
index c2950fc..3c3750b 100644
--- a/etiquette/objects.py
+++ b/etiquette/objects.py
@@ -1035,6 +1035,22 @@ class Photo(ObjectBase):
 
         self.__reinit__()
 
+    def set_override_filename(self, new_filename, *, commit=True):
+        if not new_filename:
+            new_filename = None
+        else:
+            new_filename = helpers.remove_path_badchars(new_filename)
+
+        data = {
+            'id': self.id,
+            'override_filename': new_filename,
+        }
+        self.photodb.sql_update(table='photos', pairs=data, where_key='id')
+
+        if commit:
+            self.photodb.log.debug('Committing - set override filename')
+            self.photodb.commit()
+
     def sorted_tags(self):
         tags = self.get_tags()
         tags.sort(key=lambda x: x.qualified_name())