Skip to content

Commit

Permalink
initial code for QM python api
Browse files Browse the repository at this point in the history
When integration happens, it's important to have API.
Example: creating and managing containers via python.

Signed-off-by: Douglas Schilling Landgraf <[email protected]>
  • Loading branch information
dougsland committed Jan 1, 2025
1 parent ef5463d commit 77b0bde
Show file tree
Hide file tree
Showing 3 changed files with 308 additions and 0 deletions.
69 changes: 69 additions & 0 deletions api/python/qm/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
For now just create a dir in the python latest version in your system.

1. Create a dir in the python site-packages for QM

```console
sudo mkdir -p /usr/lib/python3.13/site-packages/qm/
```
2. Copy library to it site-library for the specific user, in the case
bellow will be available only for root user.

```console
sudo cp __init__.py /usr/lib/python3.13/site-packages/qm/
```
3. Test it

Source code:

```python
from qm import QM

def main():
# Initialize the QM class
qm_manager = QM()

# Example: Start a container named 'qm'
container_name = "qm"
if qm_manager.start_container(container_name):
print(f"Container '{container_name}' started successfully.")
else:
print(f"Failed to start container '{container_name}'.")

# Example: Check the status of the container
status = qm_manager.container_status(container_name)
#print(f"Status of container '{container_name}': {status}")
print(f"Status of container {status}")

# Example: Count the number of running containers
running_count = qm_manager.count_running_containers()
print(f"Number of running containers: {running_count}")

# Example: Stop the container
if qm_manager.stop_container(container_name):
print(f"Container '{container_name}' stopped successfully.")
else:
print(f"Failed to stop container '{container_name}'.")

# Example: Check if the QM service is installed
if qm_manager.is_qm_service_installed():
print("QM service is installed.")
else:
print("QM service is not installed.")

# Example: Check if the QM service is running
if qm_manager.is_qm_service_running():
print("QM service is running.")
else:
print("QM service is not running.")

if __name__ == "__main__":
main()
```

Testing:

```console
$ sudo ./reading_information
QM service is installed.
QM service is running.
```
164 changes: 164 additions & 0 deletions api/python/qm/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,164 @@
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
A module to manage containers inside the "qm" Podman container.
This module provides a `QM` class for interacting with and managing
containers within the "qm" Podman container.
"""

import subprocess
import json


class QM:
"""Manage containers inside the 'qm' Podman container."""

def __init__(self, qm_container_name="qm"):
"""
Initialize the QM class with the name of the main Podman container.
Args:
qm_container_name (str): The name of the container running QM.
Defaults to "qm".
"""
self.qm_container_name = qm_container_name

def execute_command(self, command):
"""
Execute a generic command inside the QM container.
Args:
command (list): The command to execute inside the
QM container as a list of strings.
Returns:
str: The output of the executed command.
"""
full_command = ["podman", "exec", self.qm_container_name] + command
try:
result = subprocess.run(
full_command,
check=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
return result.stdout.decode("utf-8").strip()
except subprocess.CalledProcessError as e:
return f"Error: {e.stderr.decode('utf-8').strip()}"

def start_container(self, name):
"""
Start a container inside the QM container.
Args:
name (str): The name of the container to start.
Returns:
bool: True if the container was started successfully
"""
command = ["podman", "start", name]
output = self.execute_command(command)
return "started" in output.lower()

def stop_container(self, name):
"""
Stop a container inside the QM container.
Args:
name (str): The name of the container to stop.
Returns:
bool: True if the container was stopped successfully
"""
command = ["podman", "stop", name]
output = self.execute_command(command)
return "stopped" in output.lower()

def container_status(self, name):
"""
Check the status of a container inside the QM container.
Args:
name (str): The name of the container to check.
Returns:
str: The status of the container (e.g., "running", "stopped").
"""
command = ["podman", "inspect", "--format", "{{json .State}}", name]
output = self.execute_command(command)

if "Error" in output:
return output

try:
state = json.loads(output)
return state.get("Status", "unknown")
except json.JSONDecodeError:
return "unknown"

def count_running_containers(self):
"""
Count the number of running containers inside the QM container.
Returns:
int: The number of running containers.
"""
command = [
"podman",
"ps",
"--filter",
"status=running",
"--format",
"{{.ID}}"
]
output = self.execute_command(command)
if "Error" in output:
return 0
return len(output.splitlines()) if output else 0

def is_qm_service_installed(self):
"""
Check if the QM service is installed.
Returns:
bool: True if the QM service is installed, False otherwise.
"""
try:
result = subprocess.run(
["rpm", "-q", "qm"],
check=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
return result.returncode == 0
except subprocess.CalledProcessError:
return False

def is_qm_service_running(self):
"""
Check if the QM service is running.
Returns:
bool: True if the QM service is running, False otherwise.
"""
try:
result = subprocess.run(
["systemctl", "is-active", "qm"],
check=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
return result.stdout.decode("utf-8").strip() == "active"
except subprocess.CalledProcessError:
return False
75 changes: 75 additions & 0 deletions api/python/qm/examples/reading_information
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
#!/usr/bin/env python
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
Module for managing QM containers and checking service status.
This script demonstrates the usage of the QM class to start, stop,
and check the status of Podman containers, as well as verify the
installation and running state of the QM service.
"""

from qm import QM


def main():
"""
Demonstrate QM container and service management operations.
The main function initializes a QM instance and performs the following:
- Starts a container named 'qm'.
- Checks the container's status.
- Counts the number of running containers.
- Stops the container.
- Verifies if the QM service is installed.
- Verifies if the QM service is running.
"""
# Initialize the QM class
qm_manager = QM()

# Example: Start a container named 'qm'
container_name = "qm"
if qm_manager.start_container(container_name):
print(f"Container '{container_name}' started successfully.")
else:
print(f"Failed to start container '{container_name}'.")

# Example: Check the status of the container
status = qm_manager.container_status(container_name)
print(f"Status of container {status}")

# Example: Count the number of running containers
running_count = qm_manager.count_running_containers()
print(f"Number of running containers: {running_count}")

# Example: Stop the container
if qm_manager.stop_container(container_name):
print(f"Container '{container_name}' stopped successfully.")
else:
print(f"Failed to stop container '{container_name}'.")

# Example: Check if the QM service is installed
if qm_manager.is_qm_service_installed():
print("QM service is installed.")
else:
print("QM service is not installed.")

# Example: Check if the QM service is running
if qm_manager.is_qm_service_running():
print("QM service is running.")
else:
print("QM service is not running.")


if __name__ == "__main__":
main()

0 comments on commit 77b0bde

Please sign in to comment.