a2km 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 76ed2ec1339c1487ce7d6a76409351704bcb7189
+  data.tar.gz: 27373e1bc5980983ae56f0ef4bbe81fae6d7f6cc
+SHA512:
+  metadata.gz: 135ca36a33606c8222ab3f835a6801dbe0c16e82c43e5b29c8f59af338a9d9cf5215de70ba9bc95fbbc64339f0d10ec26ce3e0b49e0a857eee9b3c94640f6ac9
+  data.tar.gz: 7dc5323e9b547e13e1d0f8fc0afdab70cf0421dc1fd898bc4158c91be6349588970884826c1e3858708d18ca99d3e7fb0bab923924f0ba67d84a02248ce34de2

data/bin/a2km

@@ -0,0 +1,284 @@
+#!/usr/bin/env ruby
+
+require 'json'
+
+require 'rubygems'
+require 'commander'
+require 'highline'
+
+
+class AssitantToTheKernelManager
+  include Commander::Methods
+  
+  def initialize
+    @kernels = nil
+  end
+  
+  def kernels
+    if @kernels.nil?
+      js = JSON.parse `jupyter kernelspec list --json`
+      @kernels = js['kernelspecs']
+    end
+    @kernels
+  end
+  
+  def kernel_dirs
+    paths = JSON.parse `jupyter --paths --json`
+    paths['data'].map do |p|
+      p + '/' + 'kernels'
+    end
+  end
+  
+  def user_kernel_dir
+    kernel_dirs.first
+  end
+  
+  def get_kernel(name)
+    "Get a kernel by name. Die with message if not found."
+    if not kernels.has_key? name
+      STDERR.puts "No such kernel: #{name}"
+      STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
+      exit(-1)
+    end
+    kernels[name]
+  end
+  
+  def kernel_json_path(name)
+    "Return path to a kernel's kernel.json"
+    kernel = get_kernel(name)
+    File.join(kernel['resources_dir'], 'kernel.json')
+  end
+  
+  def get_kernel_json(name)
+    "Return kernel.json contents for a given kernelspec"
+    File.open(kernel_json_path(name)) do |f|
+      return JSON.parse f.read
+    end
+  end
+  
+  def save_kernel_json(name, data)
+    "save new kernel JSON"
+    File.open(kernel_json_path(name), 'w') do |f|
+      f.write JSON.pretty_generate(data)
+    end
+  end
+  
+  def run
+    program :name, 'Assistant to the KernelManager'
+    program :version, '0.0.0'
+    program :description, 'Work with Jupyter kernelspecs'
+    
+    command :rename do |c|
+      c.syntax = 'a2km rename <from> <to>'
+      c.description = 'renames kernelspec FROM to TO'
+      c.action do |args, options|
+        from = args.shift
+        to = args.shift
+        kernel = get_kernel(from)
+        src = kernel['resources_dir']
+        
+        dst = File.join File.dirname(src), to
+        if File.exists? dst
+          STDERR.puts "Destination already exists: #{dst}"
+          exit(-1)
+        end
+        
+        puts "Moving #{src} → #{dst}"
+        FileUtils.mv src, dst
+      end
+    end
+    
+    command :show do |c|
+      c.syntax = 'a2km show <spec>'
+      c.description = 'Show info about a kernelspec'
+      c.action do |args, options|
+        name = args.first
+        kernel = get_kernel(name)
+        spec = kernel['spec']
+        puts "Kernel: #{name} (#{spec['display_name']})"
+        puts "  path: #{kernel['resources_dir']}"
+        puts "  argv: #{spec['argv'].join(' ')}"
+      end
+    end
+    
+    command :locate do |c|
+      c.syntax = 'a2km locate <spec>'
+      c.description = 'Print the path of a kernelspec'
+      c.action do |args, options|
+        puts get_kernel(args.first)['resources_dir']
+      end
+    end
+    
+    command :"add-argv" do |c|
+      c.syntax = 'a2km add-argv <spec> <arg1> [arg2] ...'
+      c.description = 'Add argument(s) to a kernelspec launch command'
+      c.action do |args, options|
+        name = args.shift
+        kernelspec = get_kernel_json(name)
+        kernelspec['argv'] += args
+        save_kernel_json(name, kernelspec)
+        puts "New argv: #{kernelspec['argv'].join(' ')}"
+      end
+    end
+    
+    command :"rm-argv" do |c|
+      c.syntax = 'a2km rm-argv <spec> <arg1> [arg2] ...'
+      c.summary = 'Remove arguments from a kernelspec launch command'
+      c.description = c.summary + ". To remove args starting with '-'," \
+        " use '--'."
+      c.examples = [
+        "a2km rm-argv myspec x",
+        "a2km rm-argv myspec -- --debug",
+      ]
+      c.action do |args, options|
+        name = args.shift
+        to_remove = args
+        kernelspec = get_kernel_json(name)
+        argv = kernelspec['argv']
+        argv.reject! { |arg| to_remove.include? arg }
+        save_kernel_json(name, kernelspec)
+        puts "New argv: #{argv.join(' ')}"
+      end
+    end
+
+    command :"add-env" do |c|
+      c.syntax = 'a2km add-env <spec> <key=value> [key=value] ...'
+      c.description = 'Add environment variables to a kernelspec.' \
+        ' If no value is given, the value from the current env is used.'
+      c.action do |args, options|
+        name = args.shift
+        spec = get_kernel_json(name)
+        if not spec.has_key? 'env'
+          spec['env'] = {}
+        end
+        env = spec['env']
+        args.each do |arg|
+          key_value = arg.split('=', 2)
+          key = key_value.first
+          if key_value.length == 2
+            value = key_value[1]
+          else
+            value = ENV[key]
+          end
+          env[key] = value
+        end
+        save_kernel_json(name, spec)
+        puts "New env: #{env}"
+      end
+    end
+    
+    command :"rm-env" do |c|
+      c.syntax = 'a2km rm-env <spec> <key> [key] ...'
+      c.description = 'Remove environment variables from a kernelspec.'
+      c.action do |args, options|
+        name = args.shift
+        spec = get_kernel_json(name)
+        if not spec.has_key? 'env'
+          spec['env'] = {}
+        end
+        env = spec['env']
+        args.each do |arg|
+          env.delete(arg)
+        end
+        save_kernel_json(name, spec)
+        puts "New env: #{env}"
+      end
+    end
+
+    command :rm do |c|
+      c.syntax = 'a2km rm <spec>'
+      c.description = 'Remove a kernelspec'
+      c.option '-f', "Force removal (skip confirmation)"
+      c.action do |args, options|
+        name = args.shift
+        if not kernels.has_key? name
+          STDERR.puts "No such kernel: #{name}"
+          STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
+          exit(-1)
+        end
+        path = kernels[name]['resources_dir']
+        if options.f or not HighLine.agree("Permanently delete #{path}? (yes/no)")
+          STDERR.puts "Aborting."
+          exit(-1)
+        end
+        puts "Removing #{path}"
+        FileUtils.rm_r path
+      end
+    end
+    
+    command :set do |c|
+      c.syntax = 'a2km set <name> <key> <value>'
+      c.description = 'Set a value in the kernelspec'
+      c.examples = [
+        'a2km set python3 display_name "My Python 3"'
+      ]
+      c.action do |args, options|
+        name = args.shift
+        key = args.shift
+        value = args.shift
+        if value.nil?
+          STDERR.puts c.syntax
+          exit(-1)
+        end
+        puts "Setting #{name}.#{key} = '#{value}'"
+        spec = get_kernel_json(name)
+        spec[key] = value
+        save_kernel_json(name, spec)
+      end
+    end
+    
+    command :clone do |c|
+      c.syntax = 'a2km clone <from> <to> [display_name]'
+      c.summary = 'Clone a kernelspec'
+      c.description = 'Clones kernelspec FROM to TO'
+      c.option '--user', 'Force clone to be in the user directory' \
+        ' (default is to use the same directory as FROM)'
+      c.action do |args, options|
+        options.default :user => false
+        
+        if args.length < 2 or args.length > 3
+          STDERR.puts "Must specify FROM and TO"
+          exit(-1)
+        end
+        from_name = args.shift
+        to_name = args.shift
+        if args.length > 0
+          display_name = args.shift
+        else
+          display_name = to_name
+        end
+        
+        from = get_kernel(from_name)
+        from = kernels[from_name]
+        src = from['resources_dir']
+        
+        if options.user?
+          dst_dir = user_kernel_dir
+          makedirs(dst_dir)
+        else
+          dst_dir = File.dirname src
+        end
+        dst = File.join dst_dir, to_name
+        if File.exists? dst
+          STDERR.puts "Destination already exists: #{dst}"
+          exit(-1)
+        end
+        
+        puts "Cloning #{src} → #{dst}"
+        FileUtils.cp_r src, dst
+        kernel_json = File.join(dst, 'kernel.json')
+        kernelspec = File.open(kernel_json) do |f|
+          JSON.parse f.read
+        end
+        kernelspec['display_name'] = display_name
+        File.open(kernel_json, 'w') do |f|
+          f.write JSON.pretty_generate(kernelspec)
+        end
+      end
+    end
+
+    run!
+  end
+end
+
+AssitantToTheKernelManager.new.run

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: a2km
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Min RK
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-18 00:00:00.000000000 Z
+dependencies: []
+description: Working with Juptyer kernels
+email: benjaminrk@gmail.com
+executables:
+- a2km
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/a2km
+homepage: http://github.com/minrk/a2km
+licenses:
+- BSD
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements:
+- commander
+- highline
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Assistant to the Kernel Manager
+test_files: []
+has_rdoc: