|
5 | 5 |
|
6 | 6 | :license: MIT, see LICENSE for more details. |
7 | 7 | """ |
| 8 | +import datetime |
8 | 9 | import logging |
9 | 10 | import socket |
10 | 11 | import time |
11 | 12 |
|
12 | 13 | from SoftLayer.decoration import retry |
| 14 | +from SoftLayer import exceptions |
13 | 15 | from SoftLayer.exceptions import SoftLayerError |
14 | 16 | from SoftLayer.managers import ordering |
15 | 17 | from SoftLayer.managers.ticket import TicketManager |
|
18 | 20 | LOGGER = logging.getLogger(__name__) |
19 | 21 |
|
20 | 22 | # Invalid names are ignored due to long method names and short argument names |
21 | | -# pylint: disable=invalid-name, no-self-use |
| 23 | +# pylint: disable=invalid-name, no-self-use, too-many-lines |
22 | 24 |
|
23 | 25 | EXTRA_CATEGORIES = ['pri_ipv6_addresses', |
24 | 26 | 'static_ipv6_addresses', |
@@ -814,6 +816,145 @@ def authorize_storage(self, hardware_id, username_storage): |
814 | 816 |
|
815 | 817 | return result |
816 | 818 |
|
| 819 | + def upgrade(self, instance_id, memory=None, |
| 820 | + nic_speed=None, drive_controller=None, |
| 821 | + public_bandwidth=None, test=False): |
| 822 | + """Upgrades a hardware server instance. |
| 823 | +
|
| 824 | + :param int instance_id: Instance id of the hardware server to be upgraded. |
| 825 | + :param int memory: Memory size. |
| 826 | + :param string nic_speed: Network Port Speed data. |
| 827 | + :param string drive_controller: Drive Controller data. |
| 828 | + :param int public_bandwidth: Public keyName data. |
| 829 | + :param bool test: Test option to verify the request. |
| 830 | +
|
| 831 | + :returns: bool |
| 832 | + """ |
| 833 | + upgrade_prices = self._get_upgrade_prices(instance_id) |
| 834 | + prices = [] |
| 835 | + data = {} |
| 836 | + |
| 837 | + if memory: |
| 838 | + data['memory'] = memory |
| 839 | + if nic_speed: |
| 840 | + data['nic_speed'] = nic_speed |
| 841 | + if drive_controller: |
| 842 | + data['disk_controller'] = drive_controller |
| 843 | + if public_bandwidth: |
| 844 | + data['bandwidth'] = public_bandwidth |
| 845 | + |
| 846 | + server_response = self.get_instance(instance_id) |
| 847 | + package_id = server_response['billingItem']['package']['id'] |
| 848 | + |
| 849 | + maintenance_window = datetime.datetime.now(utils.UTC()) |
| 850 | + order = { |
| 851 | + 'complexType': 'SoftLayer_Container_Product_Order_Hardware_Server_Upgrade', |
| 852 | + 'properties': [{ |
| 853 | + 'name': 'MAINTENANCE_WINDOW', |
| 854 | + 'value': maintenance_window.strftime("%Y-%m-%d %H:%M:%S%z") |
| 855 | + }], |
| 856 | + 'hardware': [{'id': int(instance_id)}], |
| 857 | + 'packageId': package_id |
| 858 | + } |
| 859 | + |
| 860 | + for option, value in data.items(): |
| 861 | + price_id = self._get_prices_for_upgrade_option(upgrade_prices, option, value) |
| 862 | + if not price_id: |
| 863 | + # Every option provided is expected to have a price |
| 864 | + raise exceptions.SoftLayerError( |
| 865 | + "Unable to find %s option with value %s" % (option, value)) |
| 866 | + |
| 867 | + prices.append({'id': price_id}) |
| 868 | + |
| 869 | + order['prices'] = prices |
| 870 | + |
| 871 | + if prices: |
| 872 | + if test: |
| 873 | + self.client['Product_Order'].verifyOrder(order) |
| 874 | + else: |
| 875 | + self.client['Product_Order'].placeOrder(order) |
| 876 | + return True |
| 877 | + return False |
| 878 | + |
| 879 | + @retry(logger=LOGGER) |
| 880 | + def get_instance(self, instance_id): |
| 881 | + """Get details about a hardware server instance. |
| 882 | +
|
| 883 | + :param int instance_id: the instance ID |
| 884 | + :returns: A dictionary containing a large amount of information about |
| 885 | + the specified instance. |
| 886 | + """ |
| 887 | + mask = [ |
| 888 | + 'billingItem[id,package[id,keyName]]' |
| 889 | + ] |
| 890 | + mask = "mask[%s]" % ','.join(mask) |
| 891 | + |
| 892 | + return self.hardware.getObject(id=instance_id, mask=mask) |
| 893 | + |
| 894 | + def _get_upgrade_prices(self, instance_id, include_downgrade_options=True): |
| 895 | + """Following Method gets all the price ids related to upgrading a Hardware Server. |
| 896 | +
|
| 897 | + :param int instance_id: Instance id of the Hardware Server to be upgraded. |
| 898 | +
|
| 899 | + :returns: list |
| 900 | + """ |
| 901 | + mask = [ |
| 902 | + 'id', |
| 903 | + 'locationGroupId', |
| 904 | + 'categories[name,id,categoryCode]', |
| 905 | + 'item[keyName,description,capacity,units]' |
| 906 | + ] |
| 907 | + mask = "mask[%s]" % ','.join(mask) |
| 908 | + return self.hardware.getUpgradeItemPrices(include_downgrade_options, id=instance_id, mask=mask) |
| 909 | + |
| 910 | + @staticmethod |
| 911 | + def _get_prices_for_upgrade_option(upgrade_prices, option, value): |
| 912 | + """Find the price id for the option and value to upgrade. This |
| 913 | +
|
| 914 | + :param list upgrade_prices: Contains all the prices related to a |
| 915 | + hardware server upgrade. |
| 916 | + :param string option: Describes type of parameter to be upgraded |
| 917 | +
|
| 918 | + :return: int. |
| 919 | + """ |
| 920 | + price_id = None |
| 921 | + option_category = { |
| 922 | + 'memory': 'ram', |
| 923 | + 'nic_speed': 'port_speed', |
| 924 | + 'disk_controller': 'disk_controller', |
| 925 | + 'bandwidth': 'bandwidth' |
| 926 | + } |
| 927 | + category_code = option_category.get(option) |
| 928 | + |
| 929 | + for price in upgrade_prices: |
| 930 | + if price.get('categories') is None or price.get('item') is None: |
| 931 | + continue |
| 932 | + |
| 933 | + product = price.get('item') |
| 934 | + for category in price.get('categories'): |
| 935 | + if not category.get('categoryCode') == category_code: |
| 936 | + continue |
| 937 | + |
| 938 | + if option == 'disk_controller': |
| 939 | + if value == product.get('description'): |
| 940 | + price_id = price.get('id') |
| 941 | + elif option == 'nic_speed': |
| 942 | + if value.isdigit(): |
| 943 | + if str(product.get('capacity')) == str(value): |
| 944 | + price_id = price.get('id') |
| 945 | + else: |
| 946 | + split_nic_speed = value.split(" ") |
| 947 | + if str(product.get('capacity')) == str(split_nic_speed[0]) and \ |
| 948 | + split_nic_speed[1] in product.get("description"): |
| 949 | + price_id = price.get('id') |
| 950 | + elif option == 'bandwidth': |
| 951 | + if str(product.get('capacity')) == str(value): |
| 952 | + price_id = price.get('id') |
| 953 | + else: |
| 954 | + if str(product.get('capacity')) == str(value): |
| 955 | + price_id = price.get('id') |
| 956 | + |
| 957 | + return price_id |
817 | 958 |
|
818 | 959 | def _get_bandwidth_key(items, hourly=True, no_public=False, location=None): |
819 | 960 | """Picks a valid Bandwidth Item, returns the KeyName""" |
|
0 commit comments