local_model 0.1.11 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b0802a175ec198bcfbcf75e72ff213a66acd4fc7d4a9e306325e264425dd2de3
-  data.tar.gz: 96943f1468780910bb17f24d2a2fc2201b4b907e322c77a7744b2885964f8447
+  metadata.gz: 23ec22d2650b17792c2a03dff88d9dffbcedd78b08ebd05afd057fd124d0476a
+  data.tar.gz: 8c89b6eaf25891cb05c3692785d9e88979372be6078ea465d080e9fbd9a7f330
 SHA512:
-  metadata.gz: aa83e09b70e90ebdc8b379ad7dba73fd625056d0098079831a3e1f100907f3dc1ccaecde0f4c8e750c813ae8ee4a9f3d3b76e58be2127bfb6db1bd9d4f202c71
-  data.tar.gz: cd5fc6d3004722f0e4348b7321b9dcead5525abefa4fbc8720076db526ea26d98fced45b3bb79e3d3804e4a9e19457a8a9b1d3d4e4bae7c987f28a9b4c240ac0
+  metadata.gz: 6b770b1003097729755a6a63df94b909c58eb2c39ae64f398db5ff64083b2846307b173d8da3bd3533efa30ba4dce9dce2170f69ac3e6cc61d886d2b4cf08d5c
+  data.tar.gz: 405d6229dbb39b63e8fc5b9dd01d0e3dadaedf06907519f85a411a0d0547c7a1c12d6b1e7528d11f50a6d2e1cfe79f13526f5dcd6e919e1e91125ec9c03290b5

data/lib/local_model/collection.rb

@@ -21,8 +21,21 @@ class LocalModel::Collection < Array
         arg.save && self.model.save
     end
 
-    def build(arg)
-        self.push(arg)
+    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

data/lib/local_model/csv.rb

@@ -59,7 +59,7 @@ class LocalModel::CSV < LocalModel::Model
   end
 
   def self.destroy_all
-    self.all.each{ |obj| obj.destroy }
+    delete_all_rows
   end
 
   def self.where(**args)

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,24 +21,32 @@ 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)
+      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
@@ -39,15 +54,25 @@ 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
       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)
         LocalModel::Collection.create_from(
-          array: association_class.where(belongs_to_id_sym => self.id),
+          array: association_class.where(**collection_args),
           for_model: self,
           for_collection_class: association_class,
           add_to_collection_proc: add_to_collection
@@ -75,7 +100,7 @@ class LocalModel::Model
     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
@@ -85,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"
@@ -132,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.11"
+  VERSION = "0.1.12"
 end

data/lib/local_model.rb

@@ -39,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?
@@ -48,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.11
+  version: 0.1.12
 platform: ruby
 authors:
 - Micah Shute
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-06-13 00:00:00.000000000 Z
+date: 2022-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -110,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