namey 0.0.2

data/lib/namey.rb

@@ -0,0 +1,17 @@
+require 'sqlite3'
+
+module Namey
+  def self.db_path
+    @@db_path
+  end
+  def self.db_path=(x)
+    @@db_path = x
+  end
+  
+
+  # Your code goes here...
+  @@db_path = File.join(File.dirname(__FILE__), "..", "data", "names.db")
+
+  require 'namey/generator'
+end
+

data/lib/namey/generator.rb

@@ -0,0 +1,71 @@
+module Namey
+  class Generator
+
+    def initialize(dbname = Namey.db_path)     
+      @db = SQLite3::Database.open dbname
+    end
+
+    def name(frequency = :common, surname = true)
+      method = rand > 0.5 ? :male : :female
+      send(method, frequency, surname)
+    end
+    
+    def male(frequency = :common, surname = true)
+      generate(:type => :male, :frequency => frequency, :with_surname => surname)
+    end
+
+    def female(frequency = :common, surname = true)
+      generate(:type => :female, :frequency => frequency, :with_surname => surname)      
+    end
+
+    def surname(frequency = :common)
+      generate(:type => :surname, :frequency => frequency)
+    end
+    
+    def generate(params = {})
+      params = {
+        :type => :female,
+        :frequency => :common,
+        :with_surname => true
+      }.merge(params)
+
+      if ! ( params[:min_freq] || params[:max_freq] )
+        params[:min_freq], params[:max_freq] = frequency_values(params[:frequency])
+      end
+      
+      name = get_name(params[:type], params[:min_freq], params[:max_freq])
+      if params[:type] != :surname && params[:with_surname] == true
+        name = "#{name} #{get_name(:surname, params[:min_freq], params[:max_freq])}"
+      end
+      name
+    end
+    
+
+    def frequency_values(f)
+      low = case f
+            when :common then 0
+            when :rare then 40
+            when :all then 0
+            else 0
+            end
+
+      high = case f
+            when :common then 20
+            when :rare then 100
+            when :all then 100
+            else 100
+            end
+
+      [ low, high ]
+    end
+    
+    protected
+    def get_name(src, min_freq = 0, max_freq = 100)
+      @db.get_first_row("SELECT name FROM #{src.to_s} WHERE freq > ? AND freq < ? ORDER BY RANDOM() LIMIT 1;",
+                         min_freq, max_freq).first
+    end
+    
+
+    
+  end
+end

data/lib/namey/parser.rb

@@ -0,0 +1,63 @@
+module Namey
+  class Parser
+
+    attr_accessor :db
+
+    def initialize(dbname)
+      # Open a database
+      @db = SQLite3::Database.new dbname
+    end
+
+    def load_male_names(src)
+      parse_file(src, "male")
+    end
+
+    def load_female_names(src)
+      parse_file(src, "female")
+    end
+
+    def load_surnames(src)
+      parse_file(src, "surname")
+    end
+
+    protected
+    def parse_file(src, dest)
+      create_table(dest)
+
+      file = File.new(src, "r")
+      while (line = file.gets)
+        data = line.split
+
+        name = data.first.capitalize
+        freq = data[2].to_f
+
+        if dest == "surname"
+          name = cleanup_surname(name)
+        end
+        
+        puts "#{name} #{freq}"
+        @db.execute "insert into #{dest} (name, freq) values ( ?, ? )", name, freq
+      end
+      file.close     
+    end
+
+    def create_table(name)   
+      # Create a database
+      @db.execute "drop table IF EXISTS #{name};"
+      @db.execute "create table #{name} (name varchar(20), freq REAL);"
+    end
+
+    def cleanup_surname(name)
+      if name.length > 4
+        name.gsub!(/^Mc(\w+)/) { |s| "Mc#{$1.capitalize}" }
+        name.gsub!(/^Mac(\w+)/) { |s| "Mac#{$1.capitalize}" }
+        name.gsub!(/^Mac(\w+)/) { |s| "Mac#{$1.capitalize}" }
+        name.gsub!(/^Osh(\w+)/) { |s| "O'sh#{$1}" }
+        name.gsub!(/^Van(\w+)/) { |s| "Van#{$1.capitalize}" }
+        name.gsub!(/^Von(\w+)/) { |s| "Von#{$1.capitalize}" }        
+        name.gsub!(/^De(\w+)/) { |s| "De#{$1.capitalize}" }        
+      end
+      name
+    end
+  end
+end

data/lib/namey/version.rb

@@ -0,0 +1,3 @@
+module Namey
+  VERSION = "0.0.2"
+end

data/namey.gemspec

@@ -0,0 +1,39 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "namey/version"
+
+Gem::Specification.new do |s|
+  s.name        = "namey"
+  s.version     = Namey::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Colin Mitchell"]
+  s.email       = ["colin@muffinlabs.com"]
+  s.homepage    = "https://github.com/muffinista/namey"
+  s.summary     = %q{Simple name generator based on US Census Data}
+  s.description = %q{Simple name generator, which can generate male/female names based on US Census Data}
+
+  s.rubyforge_project = "namey"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<sqlite3>, [">= 0"])
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+    else
+      s.add_dependency(%q<sqlite3>, [">= 0"])
+      s.add_dependency(%q<shoulda>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<sqlite3>, [">= 0"])
+    s.add_dependency(%q<shoulda>, [">= 0"])
+  end
+end
+
+
+# http://www.census.gov/genealogy/names/

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification
+name: namey
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Colin Mitchell
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-07-13 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: &13358580 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *13358580
+- !ruby/object:Gem::Dependency
+  name: shoulda
+  requirement: &13358080 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *13358080
+description: Simple name generator, which can generate male/female names based on
+  US Census Data
+email:
+- colin@muffinlabs.com
+executables:
+- load-data
+- random-name
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/load-data
+- bin/random-name
+- data/dist.female.first
+- data/dist.male.first
+- data/names.db
+- lib/namey.rb
+- lib/namey/generator.rb
+- lib/namey/parser.rb
+- lib/namey/version.rb
+- namey.gemspec
+homepage: https://github.com/muffinista/namey
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: namey
+rubygems_version: 1.8.3
+signing_key: 
+specification_version: 3
+summary: Simple name generator based on US Census Data
+test_files: []