From 14779b5dc93276d735c98731a5d66d92dd81d235 Mon Sep 17 00:00:00 2001
From: Kim Chesed Paller <73711098+kimpaller@users.noreply.github.com>
Date: Mon, 14 Aug 2023 11:57:22 +0800
Subject: [PATCH] Update tasks.py: network

Update (network) restart_board arguments to make it the task more reconfigurable
---
 nebula/tasks.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/nebula/tasks.py b/nebula/tasks.py
index dbe6e165..92256c76 100644
--- a/nebula/tasks.py
+++ b/nebula/tasks.py
@@ -1204,13 +1204,25 @@ def check_dmesg(c, ip, user="root", password="analog", board_name=None):
         "ip": "IP address of board",
         "user": "Board username. Default: root",
         "password": "Password for board. Default: analog",
+        "yamlfilename": "Path to yaml config file. Default: /etc/default/nebula",
         "board_name": "Name of DUT design (Ex: zynq-zc706-adv7511-fmcdaq2). Require for multi-device config files",
     }
 )
-def restart_board(c, ip, user="root", password="analog", board_name=None):
+def restart_board(
+    c,
+    ip=None,
+    user=None,
+    password=None,
+    yamlfilename="/etc/default/nebula",
+    board_name=None
+):
     """Reboot development system over IP"""
     n = nebula.network(
-        dutip=ip, dutusername=user, dutpassword=password, board_name=board_name
+        dutip=ip,
+        dutusername=user,
+        dutpassword=password,
+        yamlfilename=yamlfilename,
+        board_name=board_name
     )
     n.reboot_board(bypass_sleep=True)