render_renderfarmfi/utils.py

   1 # ##### BEGIN GPL LICENSE BLOCK #####
   2 #
   3 #  This program is free software; you can redistribute it and/or
   4 #  modify it under the terms of the GNU General Public License
   5 #  as published by the Free Software Foundation; either version 2
   6 #  of the License, or (at your option) any later version.
   7 #
   8 #  This program is distributed in the hope that it will be useful,
   9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 #  GNU General Public License for more details.
  12 #
  13 #  You should have received a copy of the GNU General Public License
  14 #  along with this program; if not, write to the Free Software Foundation,
  15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16 #
  17 # ##### END GPL LICENSE BLOCK #####
  18
  19 import imp
  20
  21 from os.path import join
  22
  23 import bpy
  24
  25 from .ore_session import OreSession
  26
  27 def _write_credentials(hash, user):
  28     with open(join(bpy.utils.user_resource('CONFIG', 'rffi', True), 'rffi_credentials.py'), 'w') as pwfile:
  29         pwfile.write('hash=\''+hash+'\'\n')
  30         pwfile.write('user=\''+user+'\'')
  31
  32
  33 def _read_credentials():
  34     bpy.rffi_creds_found = False
  35     bpy.rffi_user = ''
  36     bpy.rffi_hash = ''
  37
  38     pwfile = bpy.utils.user_resource('CONFIG', 'rffi', True)
  39     try:
  40         pwmod = imp.find_module('rffi_credentials',[pwfile])
  41     except ImportError:
  42         _write_credentials('', '')
  43         pwmod = imp.find_module('rffi_credentials',[pwfile])
  44     try:
  45         user_creds = imp.load_module('rffi_credentials', pwmod[0], pwmod[1], pwmod[2])
  46         bpy.rffi_user = user_creds.user
  47         bpy.rffi_hash = user_creds.hash
  48         bpy.rffi_creds_found = True
  49     except ImportError:
  50         # doesn't exist yet, write template
  51         _write_credentials('', '')
  52         pwfile = bpy.utils.user_resource('CONFIG', 'rffi', True)
  53         pwmod = imp.find_module('rffi_credentials',[pwfile])
  54         try:
  55             user_creds = imp.load_module('rffi_credentials', pwmod[0], pwmod[1], pwmod[2])
  56             bpy.rffi_user = user_creds.user
  57             bpy.rffi_hash = user_creds.hash
  58             bpy.rffi_creds_found = True
  59         except Exception as e2:
  60             print("Couldn't write rffi_credentials.py", e2)
  61     finally:
  62         if pwmod and pwmod[0]: pwmod[0].close()
  63
  64     return bpy.rffi_creds_found
  65
  66
  67 def _xmlsessions_to_oresessions(sessions, stage=None):
  68     output = []
  69     for session in sessions:
  70         s = session['title']
  71         if stage:
  72             s = s + ' (' + stage + ')'
  73         sinfo = OreSession(session['sessionId'], s)
  74         if stage in {'Rendering', 'Completed', 'Active'}:
  75             sinfo.frames = session['framesRendered']
  76         sinfo.startframe = session['startFrame']
  77         sinfo.endframe = session['endFrame']
  78         output.append(sinfo)
  79     return output
  80
  81
  82 def update_session_list(session_list, ore):
  83     while(len(session_list) > 0):
  84         session_list.remove(0)
  85
  86     for s in bpy.ore_active_session_queue:
  87         session_list.add()
  88         session = session_list[-1]
  89         session.name = s.title + ' [' + str(s.percentageComplete()) + '% complete]'
  90
  91 def update_complete_session_list(ore):
  92     bpy.ore_active_session_queue = bpy.ore_cancelled_sessions
  93     update_session_list(ore.rejected_sessions, ore)
  94     bpy.ore_active_session_queue = bpy.ore_active_sessions
  95     update_session_list(ore.active_sessions, ore)
  96     bpy.ore_active_session_queue = bpy.ore_pending_sessions
  97     update_session_list(ore.pending_sessions, ore)
  98     bpy.ore_active_session_queue = bpy.ore_completed_sessions
  99     update_session_list(ore.completed_sessions, ore)
 100
 101     bpy.ore_complete_session_queue = []
 102     bpy.ore_complete_session_queue.extend(bpy.ore_pending_sessions)
 103     bpy.ore_complete_session_queue.extend(bpy.ore_active_sessions)
 104     bpy.ore_complete_session_queue.extend(bpy.ore_completed_sessions)
 105     bpy.ore_complete_session_queue.extend(bpy.ore_cancelled_sessions)
 106
 107     bpy.ore_active_session_queue = bpy.ore_complete_session_queue
 108     update_session_list(ore.all_sessions, ore)
 109
 110 def check_status(ore):
 111     bpy.errors = []
 112
 113     if bpy.rffi_creds_found == False and bpy.rffi_hash == '':
 114         bpy.errors.append('missing_creds')
 115
 116     if '' in {ore.title, ore.longdesc, ore.shortdesc}:
 117         bpy.errors.append('missing_desc')
 118         bpy.infoError = True
 119
 120     set_status('username', bpy.rffi_hash=='' and ore.username=='')
 121     set_status('password', bpy.rffi_hash=='' and ore.password=='')
 122
 123     set_status('title', ore.title=='')
 124     set_status('longdesc', ore.longdesc=='')
 125     set_status('shortdesc', ore.shortdesc=='')
 126
 127
 128 def set_status(property, status):
 129     if status:
 130         bpy.statusMessage[property] = 'ERROR'
 131     else:
 132         bpy.statusMessage[property] = 'TRIA_RIGHT'
 133
 134 def show_status(layoutform, property, message):
 135     if bpy.statusMessage[property] == 'ERROR':
 136         layoutform.label(text='', icon='ERROR')
 137