From bffaee539d40aa0c861e6d9d5115d9e6a7aaba53 Mon Sep 17 00:00:00 2001
From: vanzhiganov <vanzhiganov@ya.ru>
Date: Thu, 3 Nov 2016 09:08:12 +0300
Subject: [PATCH] bug fixes

---
 SWSStatisticsClient/__init__.py |  2 +-
 cloud-cron-balance.py           | 16 +++++++++++-----
 setup.py                        |  2 +-
 3 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/SWSStatisticsClient/__init__.py b/SWSStatisticsClient/__init__.py
index 437988d..836f340 100644
--- a/SWSStatisticsClient/__init__.py
+++ b/SWSStatisticsClient/__init__.py
@@ -3,7 +3,7 @@ import requests
 
 class SWSStatisticsClient(object):
     def get_vm(self, vm_id, limit=96):
-        r = requests.get('http://localhost:8089/stats/v1/compute/vms/%s' % vm_id, params={'limit': limit})
+        r = requests.get('http://server-stats.gocloud.ru/stats/v1/compute/vms/%s' % vm_id, params={'limit': limit})
         if r.status_code == 200:
             return r.json()
         return None
diff --git a/cloud-cron-balance.py b/cloud-cron-balance.py
index 9623152..fad223f 100644
--- a/cloud-cron-balance.py
+++ b/cloud-cron-balance.py
@@ -2,6 +2,8 @@
 # coding: utf-8
 
 from SWSCloudCore import models
+from SWSCloudCore.controllers.tasks import ControllerTasks
+from SWSCloudCore.controllers.vms import ControllerVMS
 
 nb = models.Settings.get_item('NEGATIVE_BALANCE')
 
@@ -16,13 +18,17 @@ if int(models.Settings.get_item('SERVICE_VMS_ENABLE')) == 1:
         # Списание средств
         x = models.UsersBalance.update(
             balance=models.UsersBalance.balance - price_quarter
-        ).where(
-            models.UsersBalance.user == vm.user.id)
+        ).where(models.UsersBalance.user == vm.user.id)
         x.execute()
 
-    # Shutting down all VMs by users who doesn't have money
-    if nb < 500:
-        pass
+        # Shutting down all VMs by users who doesn't have money
+        user_balance = models.UsersBalance.select(
+            models.UsersBalance.balance).where(models.UsersBalance.user == vm.user.id).get().balance
+        if -500 < user_balance:
+            print "user_balance", user_balance
+            ControllerVMS(vm.user.id).status_set(vm.id, 3)
+            # Создание задания
+            ControllerTasks(vm.user.id).create(vm.datacenter.id, vm.server.id, 'vm_stop', 0, vm_id=vm.id)
 
 
 if int(models.Settings.get_item('SERVICE_CONTAINERS_ENABLE')) == 1:
diff --git a/setup.py b/setup.py
index e0e63f5..949d8a8 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ from setuptools import setup
 
 setup(
     name='SWSCloudCore',
-    version='2.7.6',
+    version='2.7.7',
     author='Vyacheslav Anzhiganov',
     author_email='hello@anzhiganov.com',
     packages=[