adjust final capacity depending on publishing lambda function or not

marcy-terui · Jan 26, 2016 · d0da605 · d0da605
1 parent 129c077
commit d0da605
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/lamvery/actions.py b/lamvery/actions.py
@@ -173,14 +173,19 @@ def action(self):
             name='Function - Configuration',
             remote=remote_conf, local=local_conf, keys=CONF_DIFF_KEYS)
 
-        self._print_capacity(
-            remote=client.calculate_capacity(),
-            local=archive.get_size())
+        remote=client.calculate_capacity()
+        local=archive.get_size()
 
         if len(remote_conf) > 0:
+            if not self._publish:
+                local -= remote_conf['CodeSize']
+            self._print_capacity(remote=remote, local=local)
             client.update_function_code(zipfile, local_conf, self._publish)
             client.update_function_conf(local_conf)
         else:
+            if self._publish:
+                local *= 2
+            self._print_capacity(remote=remote, local=local)
             client.create_function(zipfile, local_conf, self._publish)
         zipfile.close()
         self._set_alias.action()