From 8816fbe54f776034af50428293b572508dda51e1 Mon Sep 17 00:00:00 2001
From: mugdhadhole1 <161293900+mugdhadhole1@users.noreply.github.com>
Date: Tue, 17 Dec 2024 12:26:58 +0530
Subject: [PATCH] Missing changes for the lobster-codebeamer supports query
 string argument (#158)

Implemented the missing changes in the lobster-codebeamer supports query
string argument

GitHub issue: #87
---
 CHANGELOG.md                           |  2 +-
 lobster/tools/codebeamer/codebeamer.py | 26 +++++++++++---------------
 2 files changed, 12 insertions(+), 16 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 367e03d..69dde53 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,7 +5,7 @@
 
 ### 0.9.21-dev
 
-* `lobster-codebeamer` now supports query string, query string (CBQL) can be passed 
+* `lobster-codebeamer` now supports query string, query string (cbQL) can be passed 
   as a command line argument to the tool `lobster-codebeamer`.
 
 * `lobster-html-report` has the following updates.
diff --git a/lobster/tools/codebeamer/codebeamer.py b/lobster/tools/codebeamer/codebeamer.py
index 3dd65e5..c94ac30 100755
--- a/lobster/tools/codebeamer/codebeamer.py
+++ b/lobster/tools/codebeamer/codebeamer.py
@@ -170,8 +170,10 @@ def get_query(mh, cb_config, query):
                         page_id,
                         cb_config["page_size"]))
         elif isinstance(query, str):
-            url = ("%s/items/query?queryString=%s" %
+            url = ("%s/items/query?page=%u&pageSize=%u&queryString=%s" %
                     (cb_config["base"],
+                        page_id,
+                        cb_config["page_size"],
                         query))
         data = query_cb_single(cb_config, url)
         assert len(data) == 4
@@ -535,20 +537,14 @@ def main():
 
     elif options.import_query:
         try:
-            if isinstance(options.import_query, int):
-                query = int(options.import_query)
-                if query < 1:
-                    ap.error("query_string must be a positive integer")
-            elif isinstance(options.import_query, str):
-                if options.import_query.startswith("-"):
-                    ap.error("query_string must be a positive integer"
-                                     " or valid string")
+            if isinstance(options.import_query, str):
+                if (options.import_query.startswith("-") and
+                    options.import_query[1:].isdigit()):
+                    ap.error("import-query must be a positive integer")
+                elif options.import_query.startswith("-"):
+                    ap.error("import-query must be a valid cbQL query")
                 elif options.import_query.isdigit():
-                    query = int(options.import_query)
-                    if query < 1:
-                        ap.error("query_string must be a positive integer")
-                else:
-                    query = str(options.import_query)
+                    options.import_query = int(options.import_query)
         except ValueError as e:
             ap.error(str(e))
 
@@ -556,7 +552,7 @@ def main():
         if options.import_tagged:
             items = import_tagged(mh, cb_config, items_to_import)
         elif options.import_query:
-            items = get_query(mh, cb_config, query)
+            items = get_query(mh, cb_config, options.import_query)
     except LOBSTER_Error:
         return 1