commit 3a9622e6c568753f9683d462de619133d8b96de1
author Drew Davenport <ddavenport@chromium.org> Wed Jun 14 09:28:25 2017 -0600
committer chrome-bot <chrome-bot@chromium.org> Thu Jun 15 11:40:45 2017 -0700
tree bfc3930c6e769cc8ae6438402ca73e8c3a77cb0c
parent c0bcabb6682eb0ad4597ee32e270d66c5633c340

netboot_firmware_settings.py: Prepend cwd to path

- Support running this tool from a symlink in factory

BUG=chromium:713599
TEST=ran platform/factory/setup/netboot_firmware_settings.py

Change-Id: Ie388d02e9144af850e447a6819af04107f52d6bf
Reviewed-on: https://chromium-review.googlesource.com/535059
Commit-Ready: Drew Davenport <ddavenport@chromium.org>
Tested-by: Drew Davenport <ddavenport@chromium.org>
Reviewed-by: Hung-Te Lin <hungte@chromium.org>

host/netboot_firmware_settings.py

diff --git a/host/netboot_firmware_settings.py b/host/netboot_firmware_settings.py
index 1add00e..4f516f9 100755
--- a/host/netboot_firmware_settings.py
+++ b/host/netboot_firmware_settings.py
@@ -9,11 +9,16 @@
 from __future__ import print_function
 
 import argparse
+import os
 import socket
 import struct
 import sys
 
 try:
+  _path = os.path.dirname(os.path.abspath(__file__))
+  if _path not in sys.path:
+    sys.path.insert(0, _path)
+  del _path
   import fmap
 except ImportError:
   print('Could not find fmap module. If you are running outside the chroot, '