123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137 |
- from pprint import pprint as pp
- import ipaddress
- import json
- import os
- class MongoHms:
- """Connect to Mongo database and provide methods to control HMS."""
- import pymongo
- def __init__(self):
- self.myclient = self.pymongo.MongoClient(
- "mongodb://"
- + "admin"
- + ":" + os.getenv("HMS_MONGO_PASSWORD")
- + "@" + ",".join(["hms01-mr.intr:27017",
- "hms02-mr.intr:27017",
- "hms03-mr.intr:27017"])
- + "/admin?replicaSet=hms-rs0"
- )
- self.service = self.myclient["staffResourceController"]["service"]
- self.socket = self.myclient["staffResourceController"]["socket"]
- self.service_socket = self.myclient["staffResourceController"]["serviceSocket"]
- self.installer = self.myclient["appscat"]["installer"]
- def ip_address_to_int(self, ip_address):
- """Convert IP address to integer."""
- return int(ipaddress.ip_address(ip_address))
- def web_find_socket(self, socket_name, server):
- """Return socket by SOCKET_NAME on SERVER."""
- return [socket for socket in self.socket.find({"name": socket_name + "@" + server})]
- def web_socket_id(self, socket_name, server):
- """Return socket_id by SOCKET_NAME on SERVER."""
- return str(self.web_find_socket(socket_name, server)[0]["_id"])
- def web_create_socket(self, socket_name, ip_address, protocol, server, port):
- """Create socket with specified SOCKET_NAME, IP_ADDRESS, PROTOCOL on SERVER PORT."""
- if not self.web_find_socket(socket_name, server):
- self.socket.insert_one({"protocol": protocol,
- "address": self.ip_address_to_int(ip_address),
- "port": port,
- "name": socket_name + "@" + server,
- "switchedOn": True,
- "_class" : "NetworkSocket"}).inserted_id
- return self.web_socket_id(socket_name, server)
- def web_service_name_to_server_id(self, service_name, server):
- """Return server id from SERVICE_NAME on SERVER."""
- return self.service.find_one({"name": service_name + "@" + server})['serverId']
- def web_find_service(self, service_name, server):
- """Return list of services matching SERVICE_NAME on SERVER."""
- return [service for service in self.service.find({"name": service_name + "@" + server})]
- def web_service_id(self, service_name, server):
- """Return service id from SERVICE_NAME on SERVER."""
- return str(self.web_find_service(service_name, server)[0]["_id"])
- def web_create_service(self, service_name, template_id, server, server_id, socket):
- """Create service with specified SERVICE_NAME, TEMPLATE_ID on SERVER with SERVER_ID on SOCKET."""
- if not self.web_find_service(service_name, server):
- self.service.insert_one({
- "serverId": server_id,
- "templateId": template_id,
- "socketIds": [socket],
- "instanceProps": {
- "security_level": "default"
- },
- "name": service_name + "@" + server,
- "switchedOn": True,
- "_class": "ru.majordomo.hms.rc.staff.resources.Service"
- })
- return self.web_service_id(service_name, server)
- def web_create_ssh_service(self, server, ip_address):
- """Create ssh service on SERVER with IP_ADDRESS."""
- return self.web_create_service(
- "ssh-guest-room",
- "5d6693a6f7619300012fbbd5",
- "5d8a2a2e708301660013776d",
- server,
- self.web_service_name_to_server_id('nginx', server),
- self.web_create_socket("ssh-guest-room-ssh", ip_address, "ssh", server, "1022")
- )
- def web_create_php_service(self, template_id, version, server, ip_address, port):
- """Create php service with specified TEMPLATE_ID with php VERSION on SERVER with IP_ADDRESS on PORT."""
- service = "apache2-php" + version + "-default"
- return self.web_create_service(
- service,
- template_id,
- server,
- self.web_service_name_to_server_id("nginx", server),
- self.web_create_socket(service + "-http", ip_address, "http", server, port),
- )
- def task_set_error(self, account_id):
- """Set all running task on ACCOUNT_ID (e.g. "208112") as processed."""
- return self.myclient["personmgr"]["processingBusinessOperation"].update_many({
- "personalAccountId":account_id,
- "state":"PROCESSING"
- }, {"$set": {"state":"ERROR"}})
- def task_set_processed(self, account_id):
- """Set all running task on ACCOUNT_ID (e.g. "208112") as processed."""
- return self.myclient["personmgr"]["processingBusinessOperation"].update_many({
- "personalAccountId":account_id,
- "state":"PROCESSING"
- }, {"$set": {"state":"PROCESSED"}})
- def wordpress(self, version):
- from datetime import datetime
- current_time = datetime.utcnow()
- self.installer.insert_one({
- "name" : f"WordPress {version} Installer",
- "dockerImage" : f"docker-registry.intr/apps/wordpress:{version}",
- "requiredParams" : [
- "DB_HOST",
- "APP_PATH",
- "APP_TITLE",
- "DB_NAME",
- "ADMIN_USERNAME",
- "DB_PASSWORD",
- "DB_USER",
- "ADMIN_EMAIL",
- "ADMIN_PASSWORD",
- "DOMAIN_NAME"
- ],
- "created" : current_time,
- "updated" : current_time,
- "_class" : "ru.majordomo.hms.appscat.model.Installer"
- })
|