zydronium/internetee-registry
1
0
Fork 0
mirror of https://github.com/internetee/registry.git synced 2025-08-04 08:52:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

Refactor account activity csv generation

Browse source
This commit is contained in:
Thiago Youssef 2022-03-31 16:33:47 +03:00
parent fd41d77390
commit daafa756aa
2 changed files with 10 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
app/services/csv_generator.rb
View file

@ -1,7 +1,7 @@
class CsvGenerator
def self.generate_csv(objects)
class_name = objects.first.class
return objects.to_csv unless custom_csv(class_name)
return objects.to_csv unless custom_csv?(class_name)
CSV.generate do |csv|
csv << class_name.csv_header
@ -9,9 +9,7 @@ class CsvGenerator
end
end
private
def self.custom_csv(class_name)
[Version::DomainVersion, Version::ContactVersion, Domain, Contact, Invoice, Account].include?(class_name)
def self.custom_csv?(class_name)
[Version::DomainVersion, Version::ContactVersion, Domain, Contact, Invoice, Account, AccountActivity].include?(class_name)
end
end