小修正

yilixiaobaisha · Aug 14, 2014 · 470a9c6 · 470a9c6
1 parent 45adc71
commit 470a9c6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/gen2.py b/gen2.py
@@ -366,9 +366,9 @@ def get_class_name( str ) :
 
 
 
-def main() :
-    filepath = r'c:/autodesk/objectarx/2012/inc-r'
 
+def generate( filepath) :
+
     #operation very line
     dir_path = filepath
     hdrfiles = []
@@ -431,5 +431,9 @@ def main() :
     print 'generate finished \n'
     return
 
+def main() :
+    filepath = r'c:/autodesk/objectarx/2012/inc-r'
+    generate(filepath)
+
 # main func 
 main()