local_model 0.1.9 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: acc6c4975bec354dd97d2a2ef5c320a9525437153b405e484970aa4542e024a7
-  data.tar.gz: dc6a281ff86af1acf3ec1a414cc75342d13466393773efe90a705f30b175ae63
+  metadata.gz: 23ec22d2650b17792c2a03dff88d9dffbcedd78b08ebd05afd057fd124d0476a
+  data.tar.gz: 8c89b6eaf25891cb05c3692785d9e88979372be6078ea465d080e9fbd9a7f330
 SHA512:
-  metadata.gz: 20d6be57b31cba2fa7f4c87a7f10235ae923e676d4a65ccc73fdf239c0c01fa757f219f3df138ffc68c41ecc6969ab266d5b8e92cd61e3b49ede6e20654c1881
-  data.tar.gz: 0ed5744beff0da65ad2aecff072d06ddf609d75203409d79c7ba934056c7d2508ae68628721bb26a5eda6ac43f76587c66fac1870ec5e10bb1b06e88947e010d
+  metadata.gz: 6b770b1003097729755a6a63df94b909c58eb2c39ae64f398db5ff64083b2846307b173d8da3bd3533efa30ba4dce9dce2170f69ac3e6cc61d886d2b4cf08d5c
+  data.tar.gz: 405d6229dbb39b63e8fc5b9dd01d0e3dadaedf06907519f85a411a0d0547c7a1c12d6b1e7528d11f50a6d2e1cfe79f13526f5dcd6e919e1e91125ec9c03290b5

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    local_model (0.1.8)
+    local_model (0.1.10)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -81,7 +81,7 @@ As of now, supports:
 - #destroy, .find, .create, .new, .where, .first, .second, .last, relationships, updating,
 
 Does not support yet (notably): 
-- .build, .update, validations, #<< to add many relationships 
+- .build, .update, validations
 - object equivalence if gotten from source twice
 
 

data/lib/local_model/collection.rb

@@ -0,0 +1,41 @@
+class LocalModel::Collection < Array
+
+    def self.create_from(array: , for_model: , for_collection_class:, add_to_collection_proc:)
+       new_obj = new(array)
+       new_obj.model = for_model 
+       new_obj.collection_class = for_collection_class
+       new_obj.add_to_collection = add_to_collection_proc
+       new_obj
+    end
+
+    attr_accessor :model, :collection_class, :add_to_collection
+
+    def <<(arg)
+        self.push(arg)
+    end
+
+    def push(arg)
+        self.[]=(self.length, arg)
+        raise ArgumentError.new("#{arg.class} inputted, expecting #{self.collection_class}") if !arg.is_a?(self.collection_class)
+        add_to_collection[arg, self.model]
+        arg.save && self.model.save
+    end
+
+    def build(**args)
+        self.push(collection_class.create(**args))
+    end
+
+    def where(**args)
+        self.filter do |el|
+            found = true
+            args.each do |k,v|
+                if el.send(k.to_s) != v
+                    found = false
+                    break
+                end
+            end
+            found
+        end
+    end
+
+end

data/lib/local_model/concerns/csv_interactable.rb

@@ -18,15 +18,15 @@ module CSVInteractable
       rand_val = @@rand.rand(1000000)
       f = File.new("#{self.storage_path}-#{rand_val}.prep", 'w')
       f.close
-      r = @@rand.rand / 10.0
-      sleep(r)
+      # r = @@rand.rand / 10.0
+      # sleep(r)
 
       if File.exist?("#{self.storage_path}.bak.csv") || Dir["#{self.storage_path}-*.prep"].length > 1
         File.delete("#{self.storage_path}-#{rand_val}.prep")
-        if @@rand.rand(2) == 1
-          sleep(0.5)
+        if @@rand.rand(5) != 1
+          sleep(@@rand.rand / 10.0)
         end
-        mutate_csv
+        mutate_csv(option, block)
         return
       else
         File.delete("#{self.storage_path}-#{rand_val}.prep")
@@ -200,6 +200,17 @@ module CSVInteractable
       end
     end
 
+    def delete_all_rows
+      begin
+        self.mutate_csv(:mutate) {}
+        true
+      rescue
+        File.delete("#{self.storage_path}.bak.csv")
+        false
+      end
+
+    end
+
   end
 
   module InstanceMethods
@@ -218,6 +229,12 @@ module CSVInteractable
       end
     end
 
+    def save!
+      if !save
+        raise LocalModel::RecordInvalid
+      end
+    end
+
   end
 
 end

data/lib/local_model/csv.rb

@@ -22,7 +22,7 @@ class LocalModel::CSV < LocalModel::Model
     end
 
     self.define_singleton_method :columns do
-        cols
+      cols
     end
 
     schema_data = cols.each_with_index.reduce({}) do |mem, (key,i)|
@@ -52,8 +52,14 @@ class LocalModel::CSV < LocalModel::Model
     all_instances
   end
 
+  def self.count
+    total = 0
+    self.each_record{ total += 1 }
+    total
+  end
+
   def self.destroy_all
-    self.all.each{ |obj| obj.destroy }
+    delete_all_rows
   end
 
   def self.where(**args)
@@ -105,12 +111,27 @@ class LocalModel::CSV < LocalModel::Model
     return self.find_by(id: id)
   end
 
+  def self.find!(id)
+    found_record = find(id)
+    if !found_record
+      raise LocalModel::RecordNotFound.new
+    else
+      found_record
+    end
+  end
+
   def self.create(**args)
     inst = new(**args)
     inst.save
     inst
   end
 
+  def self.create!(**args)
+    inst = new(**args)
+    inst.save!
+    inst
+  end
+
   def initialize(**args)
     args.each do |k,v|
       self.send("#{k}=", v)
@@ -121,6 +142,11 @@ class LocalModel::CSV < LocalModel::Model
   def saved?
     !self.id.nil?
   end
+  
+  def reload
+    raise LocalModel::RecordNotFound if !self.id
+    self.class.find!(self.id)
+  end
 
   def destroy
     self.class.delete_row({id: self.id})

data/lib/local_model/errors/record_invalid.rb

@@ -0,0 +1,4 @@
+class LocalModel::RecordInvalid < StandardError
+
+
+end

data/lib/local_model/errors/record_not_found.rb

@@ -0,0 +1,3 @@
+class LocalModel::RecordNotFound < StandardError
+
+end

data/lib/local_model/model.rb

@@ -5,7 +5,14 @@ class LocalModel::Model
     # yield(SchemaBuilder.new(self))
   end
 
-  def self.belongs_to(association, class_name: nil, foreign_key: nil)
+  def self.belongs_to(association, class_name: nil, foreign_key: nil, polymorphic: false)
+    if foreign_key.nil?
+      keyname = "#{association}_id"
+      typename = "#{association}_type"
+    else
+      keyname = foreign_key
+    end
+
     if class_name.nil?
       association_class_name = LocalModel::Functions.snake_to_camel(association)
       association_class_name[0] = association_class_name[0].upcase
@@ -14,49 +21,86 @@ class LocalModel::Model
       association_class_name = namespace_classname(class_name)
     end
 
-    if foreign_key.nil?
-      keyname = "#{association}_id"
-    else
-      keyname = foreign_key
-    end
 
     define_method association do
-      association_class = Object.const_get(association_class_name)
+      if polymorphic
+        association_type = self.send(typename)
+        return nil if association_type.nil? || association_type.empty?
+        polymorphic_class_name = LocalModel::Functions.snake_to_camel(association_type.gsub("_type", ""))
+        polymorphic_class_name[0] = polymorphic_class_name[0].upcase
+        association_class = Object.const_get(polymorphic_class_name)
+      else
+        association_class = Object.const_get(association_class_name)
+      end
       id = self.send(keyname)
       association_class.find(id)
     end
 
     define_method "#{association}=" do |association_obj|
-      self.send("#{keyname}=", association_obj.id)
+      self.send("#{keyname}=", association_obj&.id)
+      if polymorphic
+        if !association_obj.nil?
+          self.send("#{typename}=", association_obj.class.to_s)
+        end
+      end
     end
   end
 
-  def self.has_many(association, through: nil, class_name: nil, foreign_key: nil)
+  def self.has_many(association, through: nil, class_name: nil, foreign_key: nil, as: nil)
     if class_name.nil?
       association_classname = namespace_classname(get_classname_from_association(association))
     else
       association_classname = namespace_classname(class_name)
     end
 
-
     if through.nil?
-      current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(denamespace_classname(self))}_id"
+      if as.nil?
+        current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(denamespace_classname(self))}_id"
+      else
+        current_class_id_methodname = "#{as}_id"
+      end
       belongs_to_id_sym = current_class_id_methodname.to_sym
+      add_to_collection = Proc.new do |arg, model|
+        arg.send("#{belongs_to_id_sym}=", model.id)
+      end
+
       define_method association do
+        collection_args = {belongs_to_id_sym => self.id}
+        if !as.nil?
+          collection_args["#{as}_type".to_sym] = self.class.to_s
+        end
+
         association_class = Object.const_get(association_classname)
-        association_class.where(belongs_to_id_sym => self.id)
+        LocalModel::Collection.create_from(
+          array: association_class.where(**collection_args),
+          for_model: self,
+          for_collection_class: association_class,
+          add_to_collection_proc: add_to_collection
+        )
       end
     else
       current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(LocalModel::Functions.singularize(association))}_id"
       belongs_to_id_sym = current_class_id_methodname.to_sym
+      add_to_collection = Proc.new do |arg, model|
+        through_collection = model.send(through)
+        through_classname = through_collection.collection_class
+        new_join = through_classname.new
+        new_join.send("#{belongs_to_id_sym}=", arg.id)
+        through_collection << new_join
+      end
       define_method association do 
         association_class = Object.const_get(association_classname)
-        self.send(through).map{|through_obj| association_class.find(through_obj.send(belongs_to_id_sym))}
+        LocalModel::Collection.create_from(
+          array: self.send(through).map{|through_obj| association_class.find(through_obj.send(belongs_to_id_sym))},
+          for_model: self,
+          for_collection_class: association_class,
+          add_to_collection_proc: add_to_collection
+        )
       end
     end
   end
 
-  def self.has_one(association, through: nil, class_name: nil, foreign_key: nil)
+  def self.has_one(association, through: nil, class_name: nil, foreign_key: nil, as: nil)
     if class_name.nil?
       association_classname = LocalModel::Functions.snake_to_camel(association)
       association_classname[0] = association_classname[0].upcase
@@ -66,11 +110,19 @@ class LocalModel::Model
     end
 
     if through.nil?
-      current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(denamespace_classname(self))}_id"
+      if as.nil?
+        current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(denamespace_classname(self))}_id"
+      else
+        current_class_id_methodname = "#{as}_id"
+      end
       belongs_to_id_sym = current_class_id_methodname.to_sym
       define_method association do 
+        collection_args = {belongs_to_id_sym => self.id}
+        if !as.nil?
+          collection_args["#{as}_type".to_sym] = self.class.to_s
+        end
         association_class = Object.const_get(association_classname)
-        association_class.where(belongs_to_id_sym => self.id).first
+        association_class.where(**collection_args).first
       end
     else
       current_class_id_methodname = foreign_key || "#{LocalModel::Functions.camel_to_snake(association)}_id"
@@ -113,6 +165,29 @@ class LocalModel::Model
     "#{LocalModel.path}#{slash}#{self}.csv"
   end
 
+  def update(**args)
+    args.each do |k,v|
+      self.send("#{k.to_s}=", v)
+    end
+    self.save
+  end
+
+  def update!(**args)
+    args.each do |k,v|
+      self.send("#{k.to_s}=", v)
+    end
+    self.save!
+  end
+
+  def reload
+    self.class.find(self.id)
+  end
+
+  def ==(other)
+    self.class == other.class && 
+    self.id == other.id
+  end
+
 
   class SchemaBuilder
 

data/lib/local_model/version.rb

@@ -1,3 +1,3 @@
 module LocalModel
-  VERSION = "0.1.9"
+  VERSION = "0.1.12"
 end

data/lib/local_model.rb

@@ -1,6 +1,9 @@
 require_relative "./local_model/version"
 require 'csv'
+require_relative './local_model/errors/record_invalid'
+require_relative './local_model/errors/record_not_found'
 require_relative './local_model/sandbox'
+require_relative './local_model/collection'
 require_relative './local_model/adapters/boolean_adapter'
 require_relative './local_model/adapters/datetime_adapter'
 require_relative './local_model/adapters/float_adapter'
@@ -36,6 +39,13 @@ module LocalModel
     @@path
   end
 
+  def self.db_drop
+    Dir.foreach(@@path) do |f|
+      fn = File.join(@@path, f)
+      File.delete(fn) if f != '.' && f != '..'
+    end
+  end
+
   def self.config(&block)
     configuration = Configuration.new
     if block_given?
@@ -45,10 +55,7 @@ module LocalModel
     @@namespace = configuration.namespace
     Dir.mkdir(configuration.path) unless Dir.exist?(configuration.path)
     if configuration.cleanup_on_start
-      Dir.foreach(configuration.path) do |f|
-        fn = File.join(configuration.path, f)
-        File.delete(fn) if f != '.' && f != '..'
-      end
+      db_drop
     end
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: local_model
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.12
 platform: ruby
 authors:
 - Micah Shute
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-05-26 00:00:00.000000000 Z
+date: 2022-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -77,8 +77,11 @@ files:
 - lib/local_model/adapters/float_adapter.rb
 - lib/local_model/adapters/integer_adapter.rb
 - lib/local_model/adapters/string_adapter.rb
+- lib/local_model/collection.rb
 - lib/local_model/concerns/csv_interactable.rb
 - lib/local_model/csv.rb
+- lib/local_model/errors/record_invalid.rb
+- lib/local_model/errors/record_not_found.rb
 - lib/local_model/helpers/functions.rb
 - lib/local_model/helpers/pluralized_words.rb
 - lib/local_model/model.rb
@@ -107,7 +110,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.3.7
 signing_key: 
 specification_version: 4
 summary: Easily set up a schema persisted with CSV but interactable with AR-like methods