[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [25782] trunk/blender/release/scripts/io/ netrender: netrender:

Martin Poirier theeth at yahoo.com
Wed Jan 6 19:56:24 CET 2010


Revision: 25782
          http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=25782
Author:   theeth
Date:     2010-01-06 19:56:24 +0100 (Wed, 06 Jan 2010)

Log Message:
-----------
netrender:

Slave and Master options to delete files when closed (default True for slave, False for Master)
Web interface option to remove files (on master) when deleting a job (or all jobs)
Web interface button to pause a job

Modified Paths:
--------------
    trunk/blender/release/scripts/io/netrender/balancing.py
    trunk/blender/release/scripts/io/netrender/client.py
    trunk/blender/release/scripts/io/netrender/master.py
    trunk/blender/release/scripts/io/netrender/master_html.py
    trunk/blender/release/scripts/io/netrender/netrender.js
    trunk/blender/release/scripts/io/netrender/slave.py
    trunk/blender/release/scripts/io/netrender/ui.py

Modified: trunk/blender/release/scripts/io/netrender/balancing.py
===================================================================
--- trunk/blender/release/scripts/io/netrender/balancing.py	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/balancing.py	2010-01-06 18:56:24 UTC (rev 25782)
@@ -161,7 +161,7 @@
 
 class ExcludeQueuedEmptyJob(ExclusionRule):
     def __str__(self):
-        return "Exclude queued and empty jobs"
+        return "Exclude non queued and empty jobs"
 
     def test(self, job):
         return job.status != JOB_QUEUED or job.countFrames(status = QUEUED) == 0

Modified: trunk/blender/release/scripts/io/netrender/client.py
===================================================================
--- trunk/blender/release/scripts/io/netrender/client.py	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/client.py	2010-01-06 18:56:24 UTC (rev 25782)
@@ -197,7 +197,7 @@
 
         address = "" if netsettings.server_address == "[default]" else netsettings.server_address
 
-        master.runMaster((address, netsettings.server_port), netsettings.server_broadcast, netsettings.path, self.update_stats, self.test_break)
+        master.runMaster((address, netsettings.server_port), netsettings.master_broadcast, netsettings.master_clear, netsettings.path, self.update_stats, self.test_break)
 
 
     def render_slave(self, scene):

Modified: trunk/blender/release/scripts/io/netrender/master.py
===================================================================
--- trunk/blender/release/scripts/io/netrender/master.py	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/master.py	2010-01-06 18:56:24 UTC (rev 25782)
@@ -105,6 +105,17 @@
                 break
         else:
             self.status = JOB_FINISHED
+            
+    def pause(self, status = None):
+        if self.status not in {JOB_PAUSED, JOB_QUEUED}:
+            return 
+        
+        if status == None:
+            self.status = JOB_PAUSED if self.status == JOB_QUEUED else JOB_QUEUED
+        elif status:
+            self.status = JOB_QUEUED
+        else:
+            self.status = JOB_PAUSED
 
     def start(self):
         self.status = JOB_QUEUED
@@ -166,6 +177,7 @@
 log_pattern = re.compile("/log_([a-zA-Z0-9]+)_([0-9]+).log")
 reset_pattern = re.compile("/reset(all|)_([a-zA-Z0-9]+)_([0-9]+)")
 cancel_pattern = re.compile("/cancel_([a-zA-Z0-9]+)")
+pause_pattern = re.compile("/pause_([a-zA-Z0-9]+)")
 edit_pattern = re.compile("/edit_([a-zA-Z0-9]+)")
 
 class RenderHandler(http.server.BaseHTTPRequestHandler):
@@ -483,13 +495,21 @@
             match = cancel_pattern.match(self.path)
 
             if match:
+                length = int(self.headers['content-length'])
+                
+                if length > 0:
+                    info_map = eval(str(self.rfile.read(length), encoding='utf8'))
+                    clear = info_map.get("clear", False)
+                else:
+                    clear = False
+                
                 job_id = match.groups()[0]
 
                 job = self.server.getJobID(job_id)
 
                 if job:
                     self.server.stats("", "Cancelling job")
-                    self.server.removeJob(job)
+                    self.server.removeJob(job, clear)
                     self.send_head()
                 else:
                     # no such job id
@@ -497,12 +517,46 @@
             else:
                 # invalid url
                 self.send_head(http.client.NO_CONTENT)
+        # =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
+        elif self.path.startswith("/pause"):
+            match = pause_pattern.match(self.path)
 
+            if match:
+                length = int(self.headers['content-length'])
+                
+                if length > 0:
+                    info_map = eval(str(self.rfile.read(length), encoding='utf8'))
+                    status = info_map.get("status", None)
+                else:
+                    status = None
+                
+                job_id = match.groups()[0]
+
+                job = self.server.getJobID(job_id)
+
+                if job:
+                    self.server.stats("", "Pausing job")
+                    job.pause(status)
+                    self.send_head()
+                else:
+                    # no such job id
+                    self.send_head(http.client.NO_CONTENT)
+            else:
+                # invalid url
+                self.send_head(http.client.NO_CONTENT)
         # =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
         elif self.path == "/clear":
             # cancel all jobs
+            length = int(self.headers['content-length'])
+            
+            if length > 0:
+                info_map = eval(str(self.rfile.read(length), encoding='utf8'))
+                clear = info_map.get("clear", False)
+            else:
+                clear = False
+
             self.server.stats("", "Clearing jobs")
-            self.server.clear()
+            self.server.clear(clear)
 
             self.send_head()
         # =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
@@ -798,11 +852,11 @@
                     slave.job.usage += slave_usage
 
 
-    def clear(self):
+    def clear(self, clear_files = False):
         removed = self.jobs[:]
 
         for job in removed:
-            self.removeJob(job)
+            self.removeJob(job, clear_files)
 
     def balance(self):
         self.balancer.balance(self.jobs)
@@ -821,10 +875,13 @@
     def countSlaves(self):
         return len(self.slaves)
 
-    def removeJob(self, job):
+    def removeJob(self, job, clear_files = False):
         self.jobs.remove(job)
         self.jobs_map.pop(job.id)
-
+        
+        if clear_files:
+            shutil.rmtree(job.save_path)
+        
         for slave in self.slaves:
             if slave.job == job:
                 slave.job = None
@@ -856,7 +913,10 @@
 
         return None, None
 
-def runMaster(address, broadcast, path, update_stats, test_break):
+def clearMaster(path):
+    shutil.rmtree(path)
+
+def runMaster(address, broadcast, clear, path, update_stats, test_break):
         httpd = RenderMasterServer(address, RenderHandler, path)
         httpd.timeout = 1
         httpd.stats = update_stats
@@ -881,3 +941,5 @@
                         start_time = time.time()
 
         httpd.server_close()
+        if clear:
+            clearMaster(path)

Modified: trunk/blender/release/scripts/io/netrender/master_html.py
===================================================================
--- trunk/blender/release/scripts/io/netrender/master_html.py	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/master_html.py	2010-01-06 18:56:24 UTC (rev 25782)
@@ -104,7 +104,7 @@
 
         output("<h2>Master</h2>")
 
-        output("""<button title="remove all jobs" onclick="request('/clear', null);">CLEAR JOB LIST</button>""")
+        output("""<button title="remove all jobs" onclick="clear_jobs();">CLEAR JOB LIST</button>""")
 
         startTable(caption = "Rules", class_style = "rules")
 
@@ -175,7 +175,8 @@
         for job in handler.server.jobs:
             results = job.framesStatus()
             rowTable(
-                        """<button title="cancel job" onclick="request('/cancel_%s', null);">X</button>""" % job.id +
+                        """<button title="cancel job" onclick="cancel_job('%s');">X</button>""" % job.id +
+                        """<button title="pause job" onclick="request('/pause_%s', null);">P</button>""" % job.id +
                         """<button title="reset all frames" onclick="request('/resetall_%s_0', null);">R</button>""" % job.id,
                         job.id,
                         link(job.name, "/html/job" + job.id),

Modified: trunk/blender/release/scripts/io/netrender/netrender.js
===================================================================
--- trunk/blender/release/scripts/io/netrender/netrender.js	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/netrender.js	2010-01-06 18:56:24 UTC (rev 25782)
@@ -11,6 +11,28 @@
 	request("/edit_" + id, info)
 }
 
+function clear_jobs()
+{
+	var r=confirm("Also delete files on master?");
+	
+	if (r==true) {
+		request('/clear', "{'clear':True}");
+	} else {
+		request('/clear', "{'clear':False}");
+	}
+}
+
+function cancel_job(id)
+{
+	var r=confirm("Also delete files on master?");
+	
+	if (r==true) {
+		request('/cancel_' + id, "{'clear':True}");
+	} else {
+		request('/cancel_' + id, "{'clear':False}");
+	}
+}
+
 function balance_edit(id, old_value)
 {
 	var new_value = prompt("New limit", old_value);

Modified: trunk/blender/release/scripts/io/netrender/slave.py
===================================================================
--- trunk/blender/release/scripts/io/netrender/slave.py	2010-01-06 15:34:49 UTC (rev 25781)
+++ trunk/blender/release/scripts/io/netrender/slave.py	2010-01-06 18:56:24 UTC (rev 25782)
@@ -16,7 +16,7 @@
 #
 # ##### END GPL LICENSE BLOCK #####
 
-import sys, os, platform
+import sys, os, platform, shutil
 import http, http.client, http.server, urllib
 import subprocess, time
 
@@ -45,6 +45,9 @@
     def RestoreErrorMode(val):
         pass
 
+def clearSlave(path):
+    shutil.rmtree(path)
+
 def slave_Info():
     sysname, nodename, release, version, machine, processor = platform.uname()
     slave = netrender.model.RenderSlave()
@@ -103,7 +106,6 @@
             os.mkdir(NODE_PREFIX)
 
         while not engine.test_break():
-
             conn.request("GET", "/job", headers={"slave-id":slave_id})
             response = conn.getresponse()
 
@@ -162,7 +164,7 @@
                 cancelled = False
                 stdout = bytes()
                 run_t = time.time()
-                while process.poll() == None and not cancelled:
+                while not cancelled and process.poll() == None:
                     stdout += process.stdout.read(32)
                     current_t = time.time()
                     cancelled = engine.test_break()
@@ -238,10 +240,12 @@
                 for i in range(timeout):
                     time.sleep(1)
                     if engine.test_break():
-                        conn.close()

@@ Diff output truncated at 10240 characters. @@




More information about the Bf-blender-cvs mailing list