Advertisement
Guest User

Untitled

a guest
Jan 1st, 2014
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.58 KB | None | 0 0
  1. elif action.lower() == "savebuilderfiles":
  2.             self.prepareBuilderFiles(chunks[0], chunks[1], chunks[2], chunks[3], builder=True, pattern ="'builder'")    
  3.  
  4.         elif action.lower() == "savemapfiles":
  5.             self.prepareMapFiles(chunks[0], chunks[1], chunks[2], chunks[3], builder=True, pattern="'episodeofwar'")
  6.            
  7.     def prepareBuilderFiles(self, p, scenarioid, save, scenario, builder = False, pattern = None):
  8.         if not os.path.exists(p):
  9.             os.makedirs(p)
  10.  
  11.         scenarioFile = os.path.join(p, scenarioid + "_builder_scenario.lua")
  12.         saveFile = os.path.join(p, scenarioid + "_builder_save.lua")
  13.         scriptFile = os.path.join(p, scenarioid + "_builder_script.lua")
  14.        
  15.         file = QtCore.QFile(scenarioFile)
  16.         file.open(QtCore.QIODevice.WriteOnly)
  17.         file.write("version = 3")
  18.         file.write("\n")
  19.        
  20.         if builder and pattern != None:
  21.             rep = re.compile("type.*=.*('.+')")
  22.             match = re.search(rep, scenario)
  23.             if match:
  24.                 scenario.replace(match.group(1), pattern)
  25.        
  26.         file.write("ScenarioInfo" + " =" + scenario)
  27.         file.close()        
  28.        
  29.         file = QtCore.QFile(saveFile)
  30.         file.open(QtCore.QIODevice.WriteOnly)
  31.         file.write("Scenario" + " =" + save)
  32.         file.close()  
  33.  
  34.         if not os.path.exists(scriptFile):
  35.             shutil.copy2('script.lua', scriptFile)    
  36.                                
  37.     def prepareMapFiles(self, p, scenarioid, save, scenario, builder = False, pattern = None):
  38.         if not os.path.exists(p):
  39.             os.makedirs(p)
  40.  
  41.         scenarioFile = os.path.join(p, scenarioid + "_scenario.lua")
  42.         saveFile = os.path.join(p, scenarioid + "_save.lua")
  43.         scriptFile = os.path.join(p, "builder/" + scenarioid + "_builder_script.lua")
  44.         newscriptFile = os.path.join(p, scenarioid + "_script.lua")
  45.        
  46.         file = QtCore.QFile(scenarioFile)
  47.         file.open(QtCore.QIODevice.WriteOnly)
  48.         file.write("version = 3")
  49.         file.write("\n")
  50.        
  51.         if builder and pattern != None:
  52.             rep = re.compile("type.*=.*('.+')")
  53.             match = re.search(rep, scenario)
  54.             if match:
  55.                 scenario.replace(match.group(1), pattern)
  56.        
  57.         file.write("ScenarioInfo" + " =" + scenario)
  58.         file.close()        
  59.        
  60.         file = QtCore.QFile(saveFile)
  61.         file.open(QtCore.QIODevice.WriteOnly)
  62.         file.write("Scenario" + " =" + save)
  63.         file.close()  
  64.  
  65.         shutil.copy2(scriptFile, newscriptFile)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement