atk_toolbox 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0d964a5f967172531ef2a23d40319702f0f797d68aa98e8edc74388822c427e1
-  data.tar.gz: 8833017e32e64592a6631507cc28d5f7a94837030ebb5c78c3434f882fd375aa
+  metadata.gz: 3434dd3967d04310b2feefa38cc06a50afc4acd90557c3c62ac2f152037f9aa2
+  data.tar.gz: b61ada0d41d7385173f9ff0a88bab694058c7edec8fec47b681f2b0c0f70deb9
 SHA512:
-  metadata.gz: 208563db733b06e4509b198567eee63666e54b002bb858e91a4257e22e147810c11f664dd8c62359d281b6df281811ed7d27326182621bbea803c3a89f444dfc
-  data.tar.gz: 43b86e89ebf073c2046aba914667e5b76ae52c250e530b05ffa6cbfcce24234af8ab6fa8442a9120bad6c45bd33e818b8cdbf5c052fa6530e35d0b25d316be62
+  metadata.gz: 96bcc7c68e260048ae2570d3186f0c65996ab522ace3d1bb17cdd6b24235e67ff5d8dc76db49d6f1390b76a07e6adf47ece42cbf7bdf5966df9e6e565c02820b
+  data.tar.gz: b74ba108502a965b2e1353dc9a11a81fb12bed9659d27b761d572ac89e0c701d667e1d2effee01d587f7673b7958fcb5ad46aa81c46abe5e214ed3c000e92c91

data/lib/atk/os.rb

@@ -66,7 +66,7 @@ module OS
             when 'linux'
                 return (not OS.is?(:windows)) && (not OS.is?(:mac))
             when 'unix'
-                return OS.is?(:windows) || OS.is?(:mac)
+                return not( OS.is?(:windows))
         end
     end
 end

data/lib/atk/remove_indent.rb

@@ -0,0 +1,10 @@
+class String
+    # example usage 
+        # puts <<-HEREDOC.remove_indent
+        #     This is a string
+        #     this part is extra indented
+        # HEREDOC
+    def remove_indent
+        gsub(/^[ \t]{#{self.match(/^[ \t]*/)[0].length}}/, '')
+    end
+end

data/lib/atk/yaml_info_parser.rb

@@ -7,17 +7,16 @@ require 'fileutils'
 # 
 # Create loaders for ruby code literal and console code literal
 # 
+    def register_tag(tag_name, class_value)
+        YAML.add_tag(tag_name, class_value)
+        Code.tags[tag_name] = class_value
+    end
     class Code
         @@tags = {}
         def self.tags
             return @@tags
         end
         
-        def self.register_as(yaml_tag_name)
-            YAML.add_tag(yaml_tag_name, self.class)
-            Code.tags[yaml_tag_name] = self.class
-        end
-        
         def init_with(coder)
             @value = coder.scalar
         end
@@ -40,7 +39,7 @@ require 'fileutils'
             eval(@value)
         end
     end
-    RubyCode.register_as('!language/ruby')
+    register_tag('!language/ruby', RubyCode)
     # add an evaluater for ruby code
     ruby_evaluation_tag = 'evaluate/ruby'
     Code.tags[ruby_evaluation_tag] = "evaluate"
@@ -61,7 +60,7 @@ require 'fileutils'
             -"#{@value}"
         end
     end
-    ConsoleCode.register_as('!language/console')
+    register_tag('!language/console', ConsoleCode)
 # 
 # project info (specific to operating sytem)
 # 

data/lib/atk_toolbox/version.rb

@@ -0,0 +1,3 @@
+module AtkToolbox
+  VERSION = '0.0.5'
+end

data/lib/atk_toolbox.rb

@@ -1,6 +1,5 @@
-require_relative './atk/cmd'
-require_relative './atk/download'
-require_relative './atk/os'
-require_relative './atk/output'
-require_relative './atk/package_managers'
-require_relative './atk/yaml_info_parser'
+# require all the ruby files
+files = Dir.glob(File.join(__dir__, "atk", "*.rb"))
+for each in files
+    require_relative each
+end

data/lib/yaml_edit/remove_key.py

@@ -0,0 +1,29 @@
+import ruamel.yaml
+import json
+import sys
+
+yaml_string   = sys.argv[1]
+json_key_list = sys.argv[2]
+
+# a tool from https://stackoverflow.com/questions/14692690/access-nested-dictionary-path-via-a-list-of-keys
+from functools import reduce  # forward compatibility for Python 3
+import operator
+def get_by_path(data, path):
+    """Access a nested object in data by item sequence."""
+    return reduce(operator.getitem, path, data)
+
+def set_by_path(data, path, value):
+    """Set a value in a nested object in data by item sequence."""
+    get_by_path(data, path[:-1])[path[-1]] = value
+
+
+# parse the yaml
+data = ruamel.yaml.round_trip_load(yaml_string)
+# parse the path
+path_to_element = json.loads(json_key_list)
+# get the parent element
+path_element = get_by_path(data, path_to_element[:-1])
+# remove the key
+del path_element[path_to_element[-1]]
+# output the data
+print(ruamel.yaml.round_trip_dump(data))

data/lib/yaml_edit/set_key.py

@@ -0,0 +1,31 @@
+import ruamel.yaml
+import json
+import sys
+
+yaml_string   = sys.argv[1]
+json_key_list = sys.argv[2]
+new_key_value = sys.argv[3]
+
+
+# a tool from https://stackoverflow.com/questions/14692690/access-nested-dictionary-path-via-a-list-of-keys
+from functools import reduce  # forward compatibility for Python 3
+import operator
+def get_by_path(data, path):
+    """Access a nested object in data by item sequence."""
+    return reduce(operator.getitem, path, data)
+
+def set_by_path(data, path, value):
+    """Set a value in a nested object in data by item sequence."""
+    get_by_path(data, path[:-1])[path[-1]] = value
+
+
+# parse the yaml
+data = ruamel.yaml.round_trip_load(yaml_string)
+# parse the path
+path_to_element = json.loads(json_key_list)
+# parse the new value
+new_value = json.loads(new_key_value)
+# set the value
+set_by_path(data, path_to_element, new_value)
+# output the data
+print(ruamel.yaml.round_trip_dump(data))

data/lib/yaml_edit/yaml_edit.rb

@@ -0,0 +1,48 @@
+require_relative '../atk/os'
+require_relative '../atk/remove_indent'
+require 'open3'
+require 'json'
+
+def execute_with_local_python(python_file_path, *args)
+    # save the current directory
+    pwd = Dir.pwd
+    # change to where this file is
+    Dir.chdir __dir__
+    # run the python file with the virtual environment
+    stdout_str, stderr_str, status = Open3.capture3('python3', python_file_path, *args)
+    # change back to the original dir
+    Dir.chdir pwd
+    return [stdout_str, stderr_str, status]
+end
+
+def set_key(yaml_string, key_list, new_value)
+    # run the python file with the virtual environment
+    stdout_str, stderr_str, status = execute_with_local_python('set_key.py', yaml_string, key_list.to_json, new_value.to_json)
+    if not status.success?
+        raise "\n\nFailed to set key in yaml file:\n#{stderr_str}"
+    end
+    return stdout_str
+end
+
+def remove_key(yaml_string, key_list)
+    # run the python file with the virtual environment
+    stdout_str, stderr_str, status = execute_with_local_python('remove_key.py', yaml_string, key_list.to_json)
+    if not status.success?
+        raise "\n\nFailed to remove key in yaml file:\n#{stderr_str}"
+    end
+    return stdout_str
+end
+
+# yaml_test_string = <<-HEREDOC
+# foo:
+#     a: 1
+#     b: 2
+#     list:
+#         - 1
+#         - 2
+#         - 3
+# thing:
+#     a: 10
+# HEREDOC
+# puts remove_key(yaml_test_string, ["foo", "list", 2])
+# puts set_key(yaml_test_string, ["foo", "c"], 3)

data/test/main.rb

@@ -1,4 +1,4 @@
-require 'atk_toolbox'
+require_relative '../lib/atk_toolbox'
 
 Dir.chdir __dir__
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: atk_toolbox
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Jeff Hykin
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-06-06 00:00:00.000000000 Z
+date: 2019-06-07 00:00:00.000000000 Z
 dependencies: []
 description: ''
 email: jeff.hykin@gmail.com
@@ -22,8 +22,13 @@ files:
 - lib/atk/os.rb
 - lib/atk/output.rb
 - lib/atk/package_managers.rb
+- lib/atk/remove_indent.rb
 - lib/atk/yaml_info_parser.rb
 - lib/atk_toolbox.rb
+- lib/atk_toolbox/version.rb
+- lib/yaml_edit/remove_key.py
+- lib/yaml_edit/set_key.py
+- lib/yaml_edit/yaml_edit.rb
 - test/info.yaml
 - test/main.rb
 homepage: http://github.com//atk-toolbox