Esempio n. 1
0
def recompile(modulename):
    """	recompile module from modulename
	"""

    ### modulename is file type
    if os.path.exists(modulename):
        zf = ZipManager.Zip(modulename)
        return zf.Recompile()
    else:
        ### first, see if the module can be imported at all...
        tmp = __import__(modulename, globals(), locals(), fromlist=[modulename.split(".")[-1]])
        ### Use the imported module to determine its actual path
        pycfile = tmp.__file__
        modulepath = string.replace(pycfile, ".pyc", ".py")

        ### Try to open the specified module as a file
        code = open(modulepath, "rU").read()

        ### see if the file we opened can compile.  If not, return the error that it gives.
        ### if compile() fails, the module will not be replaced.
        try:
            compile(code, modulename, "exec")
        except:
            return (
                "Error in compilation: "
                + str(sys.exc_info()[0])
                + "\r\n"
                + Utilities.listf(format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback))
            )
        else:

            ### Ok, it compiled.  But will it execute without error?
            try:
                execfile(modulepath)
            except Exception:
                # return "Error '%s' happened on line %d" % (info[0], info[1][1])
                return (
                    "Error in execution: "
                    + str(sys.exc_info()[0])
                    + "\r\n"
                    + Utilities.listf(format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback))
                )

            else:

                ### at this point, the code both compiled and ran without error.  Load it up
                ### replacing the original code.
                return reload(sys.modules[modulename])
Esempio n. 2
0
def recompile(modulename):
	"""	recompile module from modulename
	"""

	### modulename is file type
	if os.path.exists(modulename):
		zf = ZipManager.Zip(modulename)
		return zf.Recompile()
	else:
		### first, see if the module can be imported at all...
		tmp = __import__(modulename, globals(), locals(), fromlist=[modulename.split('.')[-1]])
		### Use the imported module to determine its actual path
		pycfile = tmp.__file__
		modulepath = string.replace(pycfile, ".pyc", ".py")

		### Try to open the specified module as a file
		code = open(modulepath, 'rU').read()

		### see if the file we opened can compile.  If not, return the error that it gives.
		### if compile() fails, the module will not be replaced.
		try:
			compile(code, modulename, "exec")
		except:
			return "Error in compilation: " + str(sys.exc_info()[0]) + "\r\n" + Utilities.listf(format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback))
		else:

			### Ok, it compiled.  But will it execute without error?	
			try:
				execfile(modulepath)
			except Exception:
				#return "Error '%s' happened on line %d" % (info[0], info[1][1])
				return "Error in execution: " + str(sys.exc_info()[0]) + "\r\n" + Utilities.listf(format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback))

			else:

				### at this point, the code both compiled and ran without error.  Load it up
				### replacing the original code.
				return reload(sys.modules[modulename])
Esempio n. 3
0
class Zip:
	def __init__(self, fn, files=None):
		""" Constructor
		"""
		if not files: files = []
		### local copy
		self.fn = fn

		if files:
			self.Create(files)

	def Create(self, add_files=None):
		if not add_files: add_files = []
		dir_name, base_name = os.path.split(self.fn)
		name, ext = os.path.splitext(base_name)

		### output zip file
		zout = zipfile.ZipFile(self.fn, "w")

		### for all files wich could be added
		for fn in filter(lambda f: os.path.exists(f) or zipfile.is_zipfile(os.path.dirname(f)), add_files):
			fn_dir_name, fn_base_name = os.path.split(fn)
			fn_name, fn_ext = os.path.splitext(fn_base_name)
			### if adding file is compressed, we decompress and add it
			if zipfile.is_zipfile(fn_dir_name):
				zin = zipfile.ZipFile(fn_dir_name, 'r')
				buffer = zin.read(fn_base_name)
				### if not .dat file and the name of file is not the same with the zip file
				#if fn_ext == '.py':
				#zout.writestr("%s%s"%(name,fn_ext), buffer)
				#else:
				zout.writestr(fn_base_name, buffer)
				zin.close()
			else:
				zout.write(fn, fn_base_name)

		zout.close()

	def Update(self, replace_files=None):
		""" Update zip archive with the new replace file names
		"""
		if not replace_files: replace_files = []

		### delete empty fileName
		replace_files = filter(lambda f: f != '', replace_files)

		# call this function because : http://www.digi.com/wiki/developer/index.php/Error_messages
		self.ClearCache()

		zin = zipfile.ZipFile(self.fn, 'r')
		zout = zipfile.ZipFile("new_arch.zip", 'w')

		exclude_file = []

		### write all replace file in the new archive
		for fn in replace_files:
			dir_name, base_name = os.path.split(fn)
			if zipfile.is_zipfile(dir_name):
				#print '1'
				z = zipfile.ZipFile(dir_name, 'r')
				data = z.read(base_name)
				### this line is in comment because is zip file contain image file we can not encode it.
				zout.writestr(base_name, data.encode('utf-8'))
				#zout.writestr(base_name, data)
				z.close()
				#print '11'
				#sys.stdout.write("update %s from compressed %s\n"%(base_name, fn))
			elif os.path.exists(fn):
				#print '2'
				zout.write(fn, base_name)
				#print '22'
				#sys.stdout.write("update %s from %s\n"%(base_name, fn))
			elif os.path.exists(base_name) and dir_name != "":
				#print '3'
				zout.write(base_name, fn)
				#print '33'
				#sys.stdout.write("update %s from %s\n"%(fn, base_name))
			else:
				exclude_file.append(replace_files.index(fn))
				#sys.stdout.write("%s unknown\n"%(fn))

		### try to rewrite not replaced files from original zip
		info_list = zin.infolist()
		for item in info_list:
			s = os.path.basename(item.filename)
			if s not in map(os.path.basename, replace_files) and info_list.index(item) not in exclude_file:
				buffer = zin.read(item.filename)
				zout.writestr(item, buffer)
				sys.stdout.write("%s rewrite\n" % item.filename)

		### close all files
		zout.close()
		zin.close()

		### remove and rename the zip file
		self.ClearFiles()

	def Delete(self, delete_files=None):
		""" Remove file in zip archive
		"""
		if not delete_files: delete_files = []

		### delete empty fileName
		delete_files = filter(lambda f: f != '', delete_files)

		# call this function because : http://www.digi.com/wiki/developer/index.php/Error_messages
		self.ClearCache()

		zin = zipfile.ZipFile(self.fn, 'r')
		zout = zipfile.ZipFile("new_arch.zip", 'w')

		### 
		info_list = zin.infolist()
		for item in info_list:
			if item.filename not in delete_files:
				buffer = zin.read(item.filename)
				zout.writestr(item, buffer)
				##sys.stdout.write("%s rewrite\n"%(item.filename))

		### close all files
		zout.close()
		zin.close()

		### remove and rename the zip file
		self.ClearFiles()

	def GetImage(self, scaleW=16, scaleH=16):
		""" Get image object from image file stored in zip file.
			scaleH and scaleW are used to rescale image
		"""

		zf = zipfile.ZipFile(self.fn, 'r')

		### find if python file has same name of model file
		L = filter(lambda f: f.endswith(('.jpg', 'jpeg', 'png', 'bmp')), zf.namelist())

		if L:
			f = zf.open(L.pop())
			buf = f.read()
			f.close()
			sbuf = StringIO.StringIO(buf)
			image = wx.ImageFromStream(sbuf)
			image.Rescale(scaleW, scaleH)
			zf.close()
			return image
		else:
			return None

	@staticmethod
	def GetPluginFile(fn):
		""" TODO: comment
		"""
		### zipfile (amd or cmd)
		zf = zipfile.ZipFile(fn, 'r')
		nl = zf.namelist()
		zf.close()

		L = filter(lambda a: a != [], map(lambda s: re.findall("^(plugins[/]?[\w]*.py)$", s), nl))
		return L.pop(0)[0] if L != [] else ""

	@staticmethod
	def HasPlugin(fn):
		""" TODO: comment
		"""

		### zipfile (amd or cmd)
		zf = zipfile.ZipFile(fn, 'r')
		nl = zf.namelist()
		zf.close()
		### plugin file is plugins.pi in root of zipfile or in plugins zipedd directory
		return any(map(lambda s: re.search("^(plugins[/]*[\w]*.py)$", s), nl))

	# BDD Test----------------------------------------------------------------------
	@staticmethod
	def HasTests(fn):
		""" TODO: comment
		"""
		name = os.path.basename(getPythonModelFileName(fn)).split('.')[0]
		zf = zipfile.ZipFile(fn, 'r')
		nl = zf.namelist()
		zf.close()
		return any(map(lambda s: re.search("^(BDD/[\w*/]*\.py|BDD/[\w*/]*\.feature)$", s), nl))

	@staticmethod
	def GetTests(fn):
		""" Return feature, steps and environment files from .amd
		"""
		zf = zipfile.ZipFile(fn, 'r')
		nl = zf.namelist()

		zf.close()

		###
		tests_files = filter(lambda a: a != [], map(lambda s: re.findall("^(BDD/[\w*/]*\.py|BDD/[\w*/]*\.feature)$", s), nl))
		tests_files = [a[0] for a in tests_files]

		return tests_files

		#### find feature and steps file
		#feat = None
		#steps = None
		#env = None
	
		#feat = filter(lambda t: t.endswith('.feature'), zf.namelist())[0]
	
		#if 'steps.py' in zf.namelist():
			#steps = "steps.py"
		#if 'environment.py' in zf.namelist():
			#env = "environment.py"
	
		#zf.close()
	
		#return feat, steps, env
	# ------------------------------------------------------------------------------

	def GetModule(self, rcp=False):
		""" Load module from zip file corresponding to the amd or cmd model.
			It used when the tree library is created.
		"""

		# get module name
		try:
			module_name = getPythonModelFileName(self.fn)
		except Exception, info:
			sys.stderr.write(_("Error in ZipManager class for GetModule: no python file in the archive\n"))
			return info

		# if necessary, recompile (for update after editing code source of model)
		#if rcp: recompile(module_name)

		# import module
		try:
			### clear to clean the import after exporting model (amd or cmd) and reload within the same instrance of DEVSimPy
			zipimport._zip_directory_cache.clear()

			importer = zipimport.zipimporter(self.fn)
			module = importer.load_module(module_name.split('.py')[0])
			module.__name__ = Utilities.path_to_module(module_name)
		except Exception, info:
			sys.stderr.write(_("Error in execution: ") + str(sys.exc_info()[0]) + "\r\n" + Utilities.listf(
				format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback)))
			return info