From cfa762855186e030e1ab651cc1a53b0f93ebab14 Mon Sep 17 00:00:00 2001
From: j <0x006A@0x2620.org>
Date: Tue, 29 Jul 2014 17:43:38 +0200
Subject: [PATCH] reload stream from db before updating after encode, fixes
 #2442

---
 pandora/archive/models.py | 74 +++++++++++++++++----------------------
 1 file changed, 33 insertions(+), 41 deletions(-)

diff --git a/pandora/archive/models.py b/pandora/archive/models.py
index 8e705ee0d..7a4d8e451 100644
--- a/pandora/archive/models.py
+++ b/pandora/archive/models.py
@@ -624,7 +624,10 @@ class Stream(models.Model):
         return stream
 
     def path(self, name=''):
-        return self.file.get_path(name)
+        if self.source:
+            return os.path.join(os.path.dirname(self.source.media.name), name)
+        else:
+            return self.file.get_path(name)
 
     def extract_derivatives(self, rebuild=False):
         config = settings.CONFIG['video']
@@ -649,49 +652,38 @@ class Stream(models.Model):
         return True
 
     def encode(self):
-        if self.source:
-            media = self.source.media.path
-            if not self.media:
-                self.media.name = os.path.join(os.path.dirname(self.source.media.name), self.name())
-            target = self.media.path
-            info = ox.avinfo(media)
-            ok, error = extract.stream(media, target, self.name(), info)
-            if ok:
-                self.available = True
-                self.error = ''
-                if self.file.failed:
-                    self.file.failed = False
-                    self.file.save()
-            else:
-                self.media = None
-                self.available = False
-                self.error = error
-                self.file.failed = True
-                self.file.save()
-            self.save()
-        elif self.file.data:
-            media = self.file.data.path
+        media = self.source.media.path if self.source else self.file.data
+        ffmpeg = ox.file.cmd('ffmpeg')
+        if self.source or ffmpeg == 'ffmpeg':
+            ffmpeg = None
+
+        if not self.media:
+            self.media.name = self.path(self.name())
+        target = self.media.path
+
+        info = ox.avinfo(media)
+        ok, error = extract.stream(media, target, self.name(), info, ffmpeg)
+        # file could have been moved while encoding
+        # get current version from db and update
+        _self = Stream.objects.get(id=self.id)
+        _self.update_status(ok, error)
+
+    def update_status(self, ok, error):
+        if ok:
             if not self.media:
                 self.media.name = self.path(self.name())
-            target = self.media.path
-            info = ox.avinfo(media)
-            ffmpeg = ox.file.cmd('ffmpeg')
-            if ffmpeg == 'ffmpeg':
-                ffmpeg = None
-            ok, error = extract.stream(media, target, self.name(), info, ffmpeg)
-            if ok:
-                self.available = True
-                self.error = ''
-                if self.file.failed:
-                    self.file.failed = False
-                    self.file.save()
-            else:
-                self.media = None
-                self.available = False
-                self.error = error
-                self.file.failed = True
+            self.available = True
+            self.error = ''
+            if self.file.failed:
+                self.file.failed = False
                 self.file.save()
-            self.save()
+        else:
+            self.media = None
+            self.available = False
+            self.error = error
+            self.file.failed = True
+            self.file.save()
+        self.save()
 
     def make_timeline(self):
         if self.available and not self.source: