Source code for SoftLayer.managers.account

    Account manager

    :license: MIT, see License for more details.

import logging

from SoftLayer import utils

# Invalid names are ignored due to long method names and short argument names
# pylint: disable=invalid-name, no-self-use

LOGGER = logging.getLogger(__name__)

[docs]class AccountManager(utils.IdentifierMixin, object): """Common functions for getting information from the Account service :param SoftLayer.API.BaseClient client: the client instance """ def __init__(self, client): self.client = client
[docs] def get_summary(self): """Gets some basic account information :return: Account object """ mask = """mask[ nextInvoiceTotalAmount, pendingInvoice[invoiceTotalAmount], blockDeviceTemplateGroupCount, dedicatedHostCount, domainCount, hardwareCount, networkStorageCount, openTicketCount, networkVlanCount, subnetCount, userCount, virtualGuestCount ] """ return'Account', 'getObject', mask=mask)
[docs] def get_upcoming_events(self, event_type): """Retrieves a list of Notification_Occurrence_Events that have not ended yet :param: String event_type: notification event type. :return: SoftLayer_Notification_Occurrence_Event """ mask = "mask[id, subject, startDate, endDate, modifyDate, statusCode, acknowledgedFlag, " \ "impactedResourceCount, updateCount, systemTicketId, notificationOccurrenceEventType[keyName]]" _filter = { 'notificationOccurrenceEventType': { 'keyName': { 'operation': event_type } } } self.add_event_filter(_filter, event_type) return'Notification_Occurrence_Event', 'getAllObjects', filter=_filter, mask=mask, iter=True)
[docs] @staticmethod def add_event_filter(_filter, event_type): """Add data to the object filter. :param: _filter: event filter. :param: string event_type: event type. """ if event_type == 'PLANNED': _filter['endDate'] = { 'operation': '> sysdate - 2' } _filter['startDate'] = { 'operation': 'orderBy', 'options': [{ 'name': 'sort', 'value': ['DESC'] }] } if event_type == 'UNPLANNED_INCIDENT': _filter['modifyDate'] = { 'operation': '> sysdate - 2' } if event_type == 'ANNOUNCEMENT': _filter['statusCode'] = { 'keyName': { 'operation': 'in', 'options': [{ 'name': 'data', 'value': ['PUBLISHED'] }] } }
[docs] def ack_event(self, event_id): """Acknowledge an event. This mostly prevents it from appearing as a notification in the control portal. :param int event_id: Notification_Occurrence_Event ID you want to ack :return: True on success, Exception otherwise. """ return'Notification_Occurrence_Event', 'acknowledgeNotification', id=event_id)
[docs] def get_event(self, event_id): """Gets details about a maintenance event :param int event_id: Notification_Occurrence_Event ID :return: Notification_Occurrence_Event """ mask = """mask[ acknowledgedFlag, attachments, impactedResources, statusCode, updates, notificationOccurrenceEventType] """ return'Notification_Occurrence_Event', 'getObject', id=event_id, mask=mask)
[docs] def get_invoices(self, limit=50, closed=False, get_all=False): """Gets an accounts invoices. :param int limit: Number of invoices to get back in a single call. :param bool closed: If True, will also get CLOSED invoices :param bool get_all: If True, will paginate through invoices until all have been retrieved. :return: Billing_Invoice """ mask = "mask[invoiceTotalAmount, itemCount]" _filter = { 'invoices': { 'createDate': { 'operation': 'orderBy', 'options': [{ 'name': 'sort', 'value': ['DESC'] }] }, 'statusCode': {'operation': 'OPEN'}, } } if closed: del _filter['invoices']['statusCode'] return'Account', 'getInvoices', mask=mask, filter=_filter, iter=get_all, limit=limit)
[docs] def get_billing_items(self, identifier): """Gets all topLevelBillingItems from a specific invoice :param int identifier: Invoice Id :return: Billing_Invoice_Item """ mask = """mask[ id, description, hostName, domainName, oneTimeAfterTaxAmount, recurringAfterTaxAmount, createDate, categoryCode, category[name], location[name], children[id, category[name], description, oneTimeAfterTaxAmount, recurringAfterTaxAmount] ]""" return 'Billing_Invoice', 'getInvoiceTopLevelItems', id=identifier, mask=mask, iter=True, limit=100 )
[docs] def get_account_billing_items(self, mask=None): """Gets all the topLevelBillingItems currently active on the account :param string mask: Object Mask :return: Billing_Item """ if mask is None: mask = """mask[ orderItem[id,order[id,userRecord[id,email,displayName,userStatus]]], nextInvoiceTotalRecurringAmount, location, hourlyFlag ]""" object_filter = { "allTopLevelBillingItems": { "cancellationDate": { "operation": "is null" }, "createDate": utils.query_filter_orderby() } } return'Account', 'getAllTopLevelBillingItems', mask=mask, filter=object_filter, iter=True, limit=100)
[docs] def get_billing_item(self, identifier, mask=None): """Gets details about a billing item :param int identifier Billing_Item id :param string mask: Object mask to use. :return: Billing_Item """ if mask is None: mask = """mask[ orderItem[id,order[id,userRecord[id,email,displayName,userStatus]]], nextInvoiceTotalRecurringAmount, location, hourlyFlag, children ]""" return'Billing_Item', 'getObject', id=identifier, mask=mask)
[docs] def cancel_item(self, identifier, reason="No longer needed", note=None): """Cancels a specific billing item with a reason :param int identifier: Billing_Item id :param string reason: A cancellation reason :param string note: Custom note to set when cancelling. Defaults to information about who canceled the item. :return: bool """ if note is None: user ='Account', 'getCurrentUser', mask="mask[id,displayName,email,username]") note = "Cancelled by {} with the SLCLI".format(user.get('username')) return'Billing_Item', 'cancelItem', False, True, reason, note, id=identifier)