TMSS-2890: cmdline prog to call the new endpoint to remove the stations in a new copy

SAS/TMSS/client/bin/CMakeLists.txt

@@ -12,6 +12,7 @@ lofar_add_bin_scripts(tmss_get_setting)
 lofar_add_bin_scripts(tmss_set_setting)
 lofar_add_bin_scripts(tmss_populate)
 lofar_add_bin_scripts(tmss_submit_trigger)
+lofar_add_bin_scripts(tmss_remove_stations_from_scheduling_unit_blueprint)
 lofar_add_bin_scripts(tmss_wait_for_subtask_state)
 lofar_add_bin_scripts(tmss_get_subtask_l2stationspecs)
 lofar_add_bin_scripts(tmss_mark_scheduling_unit_dynamically_scheduled)

SAS/TMSS/client/bin/tmss_remove_stations_from_scheduling_unit_blueprint

+#!/usr/bin/python3
+
+# Copyright (C) 2012-2015  ASTRON (Netherlands Institute for Radio Astronomy)
+# P.O. Box 2, 7990 AA Dwingeloo, The Netherlands
+#
+# This file is part of the LOFAR software suite.
+# The LOFAR software suite is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# The LOFAR software suite is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with the LOFAR software suite. If not, see <http://www.gnu.org/licenses/>.
+
+from lofar.sas.tmss.client.mains import main_create_scheduling_unit_blueprint_copy_without_given_stations
+
+if __name__ == "__main__":
+    main_create_scheduling_unit_blueprint_copy_without_given_stations()

SAS/TMSS/client/lib/mains.py

@@ -418,3 +418,23 @@ def main_create_lofar2_sibling_scheduling_unit():
         exit(1)
 
 
+def main_create_scheduling_unit_blueprint_copy_without_given_stations():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("scheduling_unit_blueprint_id", type=int, help="The ID of the TMSS scheduling_unit_blueprint from which the station(s) have to be removed.")
+    parser.add_argument("stations_to_be_removed", type=str, help="A comma seperated string of stations.")
+    parser.add_argument('-R', '--rest_api_credentials', type=str, default='TMSSClient', help='TMSS django REST API credentials name, default: TMSSClient')
+    parser.add_argument('-r', '--retry_count', type=int, default=DEFAULT_CLIENT_RETRY_COUNT, help='Retry <retry_count> times upon (recoverable) failure, default: [%default]')
+    args = parser.parse_args()
+
+    try:
+        with TMSSsession.create_from_dbcreds_for_ldap(dbcreds_name=args.rest_api_credentials) as session:
+            stations_to_be_removed = [s.strip() for s in args.stations_to_be_removed.split(',')]
+            scheduling_unit_blueprint_copy = session.create_scheduling_unit_blueprint_copy_without_given_stations(args.scheduling_unit_blueprint_id,
+                                                                                                                  stations_to_be_removed,
+                                                                                                                  retry_count=args.retry_count)
+            print("Here is your new scheduling_unit without station(s)", ','.join(stations_to_be_removed), scheduling_unit_blueprint_copy['url'])
+    except Exception as e:
+        print(e)
+        exit(1)
+
+