genebrand 0.1.1 → 0.2.0

data/lib/genebrand/commands/new.rb

@@ -13,15 +13,23 @@ module Genebrand
             process(args, options)
           end
         end
+
+        p.default_command(:new)
       end
 
       def process(_args, _options)
         puts "Hello! Let's generate some brands!".cyan
         puts
 
+        choose do |menu|
+          menu.prompt = 'Which word DB should i use?'.yellow
+
+          menu.choice('Full (lots of words, but not high quality)') { @gen = Genebrand::Generator.new('posinfo.json') }
+          menu.choice('100k (based on top 100k English words, better quality)') { @gen = Genebrand::Generator.new('pos100k.json') }
+          menu.choice('10k (based on top 10k English words, best quality)') { @gen = Genebrand::Generator.new('pos10k.json') }
+        end
+
         brand = []
-        stopit = false
-        @gen = Genebrand::Generator.new
 
         catch :parts_done do
           loop do

data/lib/genebrand/generator.rb

@@ -4,8 +4,8 @@ module Genebrand
 
     attr_reader :words
 
-    def initialize
-      @words = JSON.parse(File.read(File.join(Gem::Specification.find_by_name('genebrand').gem_dir, 'lib/data/posinfo.json')))
+    def initialize(filename)
+      @words = JSON.parse(File.read(File.join(Gem::Specification.find_by_name('genebrand').gem_dir, "lib/data/#{filename}")))
     end
 
     def prettyoutput(domain)

data/lib/genebrand/posparser.rb

@@ -4,6 +4,10 @@ module Genebrand
     require 'fileutils'
 
     def initialize
+      init
+    end
+
+    def init
       @parsed = {}
       @table = {}
       # Сущ
@@ -19,27 +23,99 @@ module Genebrand
     end
 
     def parse(filename)
+      init
+
       unless File.exist?(filename)
         fail "File not found: #{filename}"
         return
       end
 
+      puts "Seeding"
       File.open(filename, 'r').each_line do |line|
         data = line.split("\t")
 
-        if !is_numeric?(data[0]) && (!/\A[a-zA-Z0-9]{2,10}\z/.match(data[0]).nil?)
+        data[1].split('').each do |partofsp|
+          @table[partofsp].push(data[0].downcase) if @table.key?(partofsp)
+        end
+      end
+
+      @parsed
+    end
+
+    def parse_top(filename, top)
+      init
+
+      unless File.exist?(filename)
+        fail "File not found: #{filename}"
+        return
+      end
+
+      puts "Load top"
+      toparr = []
+      File.open(top, 'r').each_line do |line|
+        toparr << line.strip.downcase
+      end
+      puts toparr.count
+
+      puts "Seeding"
+      it = 0
+      File.open(filename, 'r').each_line do |line|
+        data = line.split("\t")
+
+        if toparr.include?(data[0])
           data[1].split('').each do |partofsp|
-            @table[partofsp].push(data[0]) if @table.key?(partofsp)
+            @table[partofsp].push(data[0].downcase) if @table.key?(partofsp)
           end
         end
+        it+=1
+        if it % 10000 == 0
+          puts it
+        end
       end
 
       @parsed
     end
 
-    def parseandsave(filename)
+    def preseed(filename)
+      init
+
+      prsdata = []
+
+      unless File.exist?(filename)
+        fail "File not found: #{filename}"
+        return
+      end
+
+      puts "Preseed"
+      File.open(filename, 'r').each_line do |line|
+        data = line.split("\t")
+
+        if !is_numeric?(data[0]) && (!/\A[a-zA-Z0-9]{2,10}\z/.match(data[0]).nil?)
+          prsdata << line
+        end
+      end
+
+      prsdata
+    end
+
+    def parseandsave(filename, to)
+      FileUtils.mkdir_p 'lib/data'
+      File.open(to, 'w+') { |f| f.write(parse(filename).to_json) }
+    end
+
+    def parseandsave_preseed(filename, to)
+      FileUtils.mkdir_p 'lib/data'
+      File.open(to, 'w+') do |f|
+        write = preseed(filename)
+        write.each do |line|
+          f.write(line)
+        end
+      end
+    end
+
+    def parseandsave_top(filename, top, to)
       FileUtils.mkdir_p 'lib/data'
-      File.open('lib/data/posinfo.json', 'w+') { |f| f.write(parse(filename).to_json) }
+      File.open(to, 'w+') { |f| f.write(parse_top(filename, top).to_json) }
     end
 
     def is_numeric?(obj)

data/lib/genebrand/version.rb

@@ -1,3 +1,3 @@
 module Genebrand
-  VERSION = '0.1.1'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: genebrand
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrey Viktorov
@@ -80,20 +80,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '10.0'
-- !ruby/object:Gem::Dependency
-  name: rspec
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.2'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.2'
 description: CLI brand names generator
 email:
 - andv@outlook.com
@@ -113,6 +99,8 @@ files:
 - bin/setup
 - exe/genebrand
 - genebrand.gemspec
+- lib/data/pos100k.json
+- lib/data/pos10k.json
 - lib/data/posinfo.json
 - lib/genebrand.rb
 - lib/genebrand/command.rb
@@ -134,7 +122,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.9.3
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="