Merge pull request #1546 from flyingcys/dev_wifi

update gen_bin.py

1 files changed, 25 insertions, 5 deletions

diff --git a/bsp/amebaz/gen_bin.py b/bsp/amebaz/gen_bin.py
index 4537534f5f..36f83184e9 100644
--- a/bsp/amebaz/gen_bin.py
+++ b/bsp/amebaz/gen_bin.py
@@ -3,7 +3,27 @@ import subprocess
 import os
 import platform
 
-tool_pwd=os.path.normpath(os.getcwd()+r'/packages/realtek_ameba/sdk-ameba-v4.0b_without_NDA_GCC_V1.0.0/component/soc/realtek/8711b/misc/iar_utility/common/tools')
+#tool_pwd=os.path.normpath(os.getcwd()+r'/packages/realtek_ameba/sdk-ameba-v4.0b_without_NDA_GCC_V1.0.0/component/soc/realtek/8711b/misc/iar_utility/common/tools')
+
+pwd=[]
+def findfile(path,filename):
+    for x in os.listdir(path):
+    	if x=='8195a':
+    		continue
+        x=os.path.join(path,x)
+        if os.path.isdir(x):
+            findfile(x,filename)
+        else:
+            if filename in os.path.split(x)[1]:
+                pwd.append(os.path.split(x)[0])
+                break
+    return
+
+findfile(os.path.abspath('.'),'pick.exe')
+if pwd:
+	print(pwd[0])
+
+tool_pwd=str(pwd[0])
 
 def grep_func(file,str):
 	fd=open(file,'r')
@@ -34,14 +54,14 @@ addr6=grep_func(r'application.nmap','__xip_image2_start__')
 sysstr = platform.system()
 print(sysstr)
 if(sysstr == "Linux"):
-	exec_pick = tool_pwd + r'/pick'
-	exec_checksum = tool_pwd + r'/checksum'
+	exec_pick=os.path.join(tool_pwd,'pick')
+	exec_checksum=os.path.join(tool_pwd,'checksum')
 	os.system('chmod +x ' + exec_pick)
 	os.system('chmod +x ' + exec_checksum)
 	#print(exec_pick)
 elif(sysstr == "Windows"):
-	exec_pick=tool_pwd+r'/pick.exe'
-	exec_checksum = tool_pwd + r'/checksum.exe'
+	exec_pick=os.path.join(tool_pwd,'pick.exe')
+	exec_checksum=os.path.join(tool_pwd,'checksum.exe')
 	#print(exec_pick)
 else:
 	print("not support!!!")