seedy 0.6.3

data/lib/seedy/generators/user.rb

@@ -0,0 +1,7 @@
+module Seedy
+  module Generators
+    def user_hash
+      "meh"
+    end
+  end
+end

data/lib/seedy/generators/website.rb

@@ -0,0 +1,7 @@
+module Seedy
+  module Generators
+    def website
+      Faker::Internet.domain_name
+    end
+  end
+end

data/lib/seedy/graph.rb

@@ -0,0 +1,3 @@
+require 'seedy/graphs/abstract_graph'
+require 'seedy/graphs/simple_graph'
+

data/lib/seedy/graphs/abstract_graph.rb

@@ -0,0 +1,143 @@
+module Seedy
+  class AbstractGraph
+    @@related_records = {
+      :teams => 0,
+      :accounts => 0,
+      :bugs => 0,
+      :calls => 0,
+      :cases => 0,
+      :contacts => 0,
+     	:documents=> 0,
+      :emails => 0,    
+     	:email_addresses => 0,
+      :leads => 0,
+      :meetings => 0,
+      :notes => 0,
+      :opportunities => 0,
+      :products => 0,
+      :product_bundles => 0,
+      :quotes => 0,
+      :tasks => 0,
+    }    
+    
+    @@records = {}
+    @@buffer  = Seedy::AbstractBuffer
+    
+    # Adds an ID to the id array
+    def add(mod,id)
+      @@records[mod] = [] unless @@records[mod].class == Array
+      @@records[mod] << id
+    end
+    
+    # Selects a random ID from the id array for a given module
+    def select(mod)
+      @@records[mod][rand(@@records[mod].length - 1)]
+    end
+    
+    def records
+      @@related_records
+    end
+    
+    # Our buffer
+    def buffer
+      @@buffer
+    end
+    
+    def create_user
+      user = Users.build
+      buffer << user
+      add(:users, user)
+    end
+    
+    def create_teams
+      mod = :teams
+      records[mod].times {        
+        team = Teams.build(select(:users))
+        buffer << team
+        add(mod, team.id)
+      }
+    end
+    
+    def create_accounts
+      mod = :accounts
+      records[mod].times {        
+        account = Accounts.build(select(:users),select(:teams))
+        buffer << account
+        add(mod, account.id)
+      }      
+    end
+    
+    def create_bugs
+      mod = :bugs
+      records[mod].times {        
+        bug = Bugs.build(select(:users),select(:teams), select(:accounts))
+        buffer << bug
+        add(mod, bug.id)
+      }      
+    end
+
+    def create_contacts
+      mod = :contacts
+      records[mod].times {        
+        contact = Contacts.build(select(:users),select(:teams), select(:accounts))
+        buffer << contact
+        add(mod, contact.id)
+      }      
+    end
+    
+    def create_cases
+      mod = :cases
+      records[mod].times {        
+        cases = Cases.build(select(:users),select(:teams), select(:accounts))
+        buffer << cases
+        add(mod, cases.id)
+      }      
+    end
+    
+    def create_opportunities
+      mod = :opportunities
+      records[mod].times {        
+        opp = Opportunities.build(select(:users),select(:teams), select(:accounts))
+        buffer << opp
+        add(mod, opp.id)
+      }      
+    end
+    
+    def create_leads
+      mod = :leads
+      records[mod].times {        
+        lead = Leads.build(select(:users),select(:teams), select(:contacts), select(:opportunities), select(:accounts))
+        buffer << lead
+        add(mod, lead.id)
+      }      
+    end
+      
+    def create_calls
+      mod = :calls
+      records[mod].times {        
+        call = Calls.build(select(:users),select(:teams), select(:contacts))
+        buffer << call
+        add(mod, call.id)
+      }      
+    end
+    
+    def create_emails
+      mod = :emails
+      records[mod].times {        
+        email = Emails.build(select(:users),select(:teams))
+        buffer << email
+        add(mod, email.id)
+      }      
+    end
+    
+    def create_email_addresses
+      mod = :email_addresses
+      records[mod].times {        
+        email = EmailAddresses.build()
+        buffer << email
+        add(mod, email.id)
+      }      
+    end      
+
+  end
+end

data/lib/seedy/graphs/simple_graph.rb

@@ -0,0 +1,36 @@
+module Seedy
+class SimpleGraph < AbstractGraph
+  @@related_records = {
+    :teams => 1,
+    :accounts => 20,
+    :bugs => 60,
+    :cases => 80,
+    :contacts => 80,
+    :opportunities => 40,
+    :calls => 480,
+    :leads => 80,
+    :emails => 320,
+    :email_addresses => 80
+  }
+  
+  
+  
+  @@buffer  = Seedy::AbstractBuffer
+  
+  # Creates a new user record, and all the dependent records listed above
+  def initialize()    
+    create_user
+    create_teams
+    create_accounts
+    create_bugs
+    create_contacts
+    create_cases
+    create_opportunities
+    create_calls
+    create_leads
+    Seedy::DatabaseBuffer.flush
+  end
+  
+end
+end
+

data/lib/seedy/graphs/user_graph.rb

@@ -0,0 +1,35 @@
+module Seedy
+class UserGraph < AbstractGraph
+  @@related_records = {
+    'Teams' => 1,
+    'Accounts' => 20,
+    'Bugs' => 60,
+    'Calls' => 480,
+    'Cases' => 80,
+    'Contacts' => 80,
+   	'Documents'=> 80,
+    'Emails' => 320,    
+   	'EmailAddresses'=> 80,
+    'Leads' => 80,
+    'Meetings' => 160,
+    'Notes' => 80,
+    'Opportunities' => 40,
+    'Products' => 80,
+    'ProductBundles' => 40,
+    'Quotes' => 20,
+    'Tasks' => 80,
+  }
+  
+  # Creates a new user record, and all the dependent records listed above
+  def initialize 
+    user = Users.build
+    @@related_records.each_pair do |module_name,count|
+      count.times { 
+        record = module_name.build(self)
+        DatabaseBuffer.insert(record)
+      }
+    end
+  end
+  
+end
+end

data/lib/seedy/random.rb

@@ -0,0 +1,7 @@
+module Seedy
+  class Random    
+    Random.class_eval do
+      extend Seedy::Generators
+    end
+  end
+end

data/lib/seedy/record.rb

@@ -0,0 +1,13 @@
+require 'seedy/records/attribute_methods'
+require 'seedy/records/record'
+require 'seedy/records/accounts'
+require 'seedy/records/bugs'
+require 'seedy/records/calls'
+require 'seedy/records/cases'
+require 'seedy/records/contacts'
+require 'seedy/records/email_addresses'
+require 'seedy/records/emails'
+require 'seedy/records/leads'
+require 'seedy/records/opportunities'
+require 'seedy/records/teams'
+require 'seedy/records/users'

data/lib/seedy/records/accounts.rb

@@ -0,0 +1,49 @@
+module Seedy
+  class Accounts < Record
+    self.columns = {
+      :id => :guid,
+      :name => :company,
+      :date_entered => :date_time,
+      :date_modified => :date_time,
+      :modified_user_id => :association,
+      :created_by => :association,
+      :description => :description,
+      :deleted => 0,
+      :assigned_user_id => :association,
+      :team_id => :association,
+      :account_type => :customer_type,
+      :industry => :industry,
+      :annual_revenue => :monetary_sum,
+      :phone_fax => :phone,
+      :billing_address_street => :address_street,
+      :billing_address_city => :city,
+      :billing_address_state => :state,
+      :billing_address_postalcode => :zip,
+      :billing_address_country => :country,
+      :phone_office => :phone,
+      :website => :website,
+      :ownership => :ownership,
+      :employees => :employees,
+      :shipping_address_street => :address_street,
+      :shipping_address_city => :city,
+      :shipping_address_state => :state,
+      :shipping_address_postalcode => :zip,
+      :shipping_address_country => :country,
+    }
+
+    class << self
+      def build(user, team)
+        account = Accounts.new
+        account.associate(user,team)
+        account
+      end
+    end
+    
+    def associate(user, team)
+      write_attribute(:modified_user_id, user.id)
+      write_attribute(:assigned_user_id, user.id)
+      write_attribute(:team_id, team)
+      write_attribute(:created_by, user.id)
+    end
+  end 
+end

data/lib/seedy/records/attribute_methods.rb

@@ -0,0 +1,121 @@
+module Seedy
+  module AttributeMethods
+    # Wrapper around attributes hash
+    def read_attribute(key)
+      @attributes[key]
+    end
+
+    # Wrapper around attributes hash
+    def write_attribute(key, value)
+      @attributes[key] = value
+    end
+    
+    def table_name
+      self.class.to_s.tableize.split(/\//)[-1]
+    end
+    
+    def table_columns#:nodoc:
+      sql = "SHOW FIELDS FROM #{table_name}"
+      columns = []
+      result = execute(sql)
+      result.each { |field| columns << field[0]}
+      result.free
+      columns
+    end
+    
+    # Adds a row to a join table
+    def join(table, hash)
+      sql = <<-EOF
+      INSERT INTO #{table}
+        (#{hash.keys.join(",")})
+        VALUES
+        (#{quote(hash.values).join(",")});
+      EOF
+      sql.gsub!(/^\s{6}/, '')
+      execute(sql)
+    end
+    
+    def quote(object)
+      case object
+      when String:
+        return "'#{object}'"
+      when Fixnum:
+        return object
+      when Array:
+        objs = []
+        object.each do |o|
+          objs << quote(o)
+        end
+        return objs
+      else
+        return "'#{object}'"
+      end
+    end
+    
+    def execute(sql)
+      puts sql
+      Seedy.connection.query(sql) 
+    end
+    
+    # Checks to see if we have all the neccessary attributes
+    def valid?
+      clear_errors
+      valid = true
+      self.class.columns.keys.each do |attribute|
+        if @attributes[attribute].blank?
+          @errors.add "#{attribute} cannot be blank"
+          valid = false
+        end
+      end
+      valid
+    end
+    
+    def errors
+      @errors.to_a.join(", ")
+    end
+    
+    def clear_errors
+      @errors = Set.new
+    end
+    
+    # Generates get/set methods for keys in the attributes hash
+    def define_attribute_methods
+      return if self.class.attribute_methods_generated
+      table_columns.each do |k|
+        self.class.module_eval %Q?
+        def #{k}
+          read_attribute :#{k}
+        end
+        def #{k}=(value)
+          write_attribute :#{k},value
+        end
+        ?
+      end
+      self.class.attribute_methods_generated = true
+    end
+    
+    def generate_attributes_for(columns)
+      columns.each_pair do |column, generator|
+        case generator
+        when Symbol:
+          next if generator == :association
+          write_attribute(column, Seedy::Random.send(generator))
+        else
+          write_attribute(column, generator)
+        end
+      end
+    end
+    
+    def attribute_for_inspect(attr_name)
+      value = read_attribute(attr_name)
+      if value.is_a?(String) && value.length > 50
+        "#{value[0..50]}...".inspect
+      elsif value.is_a?(Date) || value.is_a?(Time)
+        %("#{value.to_s}")
+      else
+        value.inspect
+      end
+    end
+    
+  end
+end

data/lib/seedy/records/bugs.rb

@@ -0,0 +1,38 @@
+module Seedy
+  class Bugs < Record
+    self.columns = {
+      :id => :guid,
+      :name => :subject,
+      :date_entered => :date_time,
+      :date_modified => :date_time,
+      :modified_user_id => :association,
+      :created_by => :association,
+      :description => :description,
+      :deleted => 0,
+      :assigned_user_id => :association,
+      :team_id => :association,
+      :status => :status,
+      :priority => :priority,
+    }
+
+    class << self
+      def build(user, team, account)
+        bug = Bugs.new
+        bug.associate(user,team,account)
+        bug
+      end
+    end
+    
+    def associate(user, team, account)
+      write_attribute(:modified_user_id, user.id)
+      write_attribute(:assigned_user_id, user.id)
+      write_attribute(:team_id, team)
+      write_attribute(:created_by, user.id)
+      join("accounts_bugs", {
+        :id => Random.guid, 
+        :account_id => account, 
+        :bug_id => read_attribute(:id), 
+        :date_modified => read_attribute(:date_modified)})
+    end
+  end 
+end