DEVSimPy/Mixins/Plugable.py at a516c1d3fc00d224792a1eea8b6ea6191036fb3a · capocchi/DEVSimPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# -*- coding: utf-8 -*-

## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ##
# Plugable.py ---
#                     --------------------------------
#                        Copyright (c) 2013
#                       Laurent CAPOCCHI
#                      University of Corsica
#                     --------------------------------
# Version 1.0                                        last modified: 19/11/13
## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ##
#
# GENERAL NOTES AND REMARKS:
#
## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ##

## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ##
#
# GLOBAL VARIABLES AND FUNCTIONS
#
## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ## ##-

import sys
import os
import zipimport
import inspect
import types
import zipfile

import Decorators

class Plugable:
	""" Plugable Mixin.
	"""

	@staticmethod
	def Load_Module(fileName:str):
		""" Load module without load_module from importer.
			In this way, we can change the name of module in the built-in.
		"""

		### import zipfile model
		if zipfile.is_zipfile(fileName):
			importer = zipimport.zipimporter(fileName)

			### change module name
			old_plugin_name = 'plugins'
			new_plugin_name = '%s.%s'%(os.path.basename(os.path.splitext(fileName)[0]), old_plugin_name)

			### get code of plug-ins
			code =  importer.get_code(old_plugin_name)

			# Create the new 'temp' module.
			temp = types.ModuleType(new_plugin_name)
			sys.modules[new_plugin_name] = temp

			### there is syntax error ?
			try:
				exec(code, temp.__dict__)
			except Exception as info:
				return info

			return sys.modules[new_plugin_name]

		return None

	@Decorators.BuzyCursorNotification
	def LoadPlugins(self, fileName:str):
		""" Method which load plug-ins from zip.
			Used for define or redefine method of amd. and .cmd model.
			The name of plug-in file must be "plugins.py".
		"""

		### if list of activated plug-ins is not empty
		if self.plugins:
			module = Plugable.Load_Module(fileName)

			if inspect.ismodule(module):

				for name,m in inspect.getmembers(module, inspect.isfunction):
					### import only plug-ins in plug-ins list (dynamic attribute) and only method
					if name in self.plugins and 'self' in inspect.getargspec(m).args:
						setattr(self, name, types.MethodType(m, self, self.__class__))
			else:
				return module
		### restore method which was assigned to None before being pickled
		else:
			### for all method in the class of model
			for method in [value for value in list(self.__class__.__dict__.values()) if isinstance(value, types.FunctionType)]:
				name = method.__name__
				### if method was assigned to None by getstate before being pickled
				if getattr(self, name) is None:
					### assign to default class method
					setattr(self, name, types.MethodType(method, self))

		return True

def main():
    pass

if __name__ == '__main__':
    main()