mirror of
https://github.com/cliffe/SecGen.git
synced 2026-02-22 19:58:03 +00:00
organisations: added intro_paragraph parameter
This commit is contained in:
@@ -12,6 +12,7 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
attr_accessor :manager
|
||||
attr_accessor :employees
|
||||
attr_accessor :product_name
|
||||
attr_accessor :intro_paragraph
|
||||
|
||||
def initialize
|
||||
super
|
||||
@@ -26,6 +27,7 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
self.manager = {}
|
||||
self.employees = []
|
||||
self.product_name = ''
|
||||
self.intro_paragraph = ''
|
||||
end
|
||||
|
||||
def encode_all
|
||||
@@ -40,6 +42,7 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
organisation_hash['manager'] = self.manager
|
||||
organisation_hash['employees'] = self.employees
|
||||
organisation_hash['product_name'] = self.product_name
|
||||
organisation_hash['intro_paragraph'] = self.intro_paragraph
|
||||
|
||||
self.outputs << organisation_hash.to_json
|
||||
end
|
||||
@@ -54,7 +57,8 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
['--industry', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--manager', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--employees', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--product_name', GetoptLong::REQUIRED_ARGUMENT]]
|
||||
['--product_name', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--intro_paragraph', GetoptLong::REQUIRED_ARGUMENT]]
|
||||
end
|
||||
|
||||
def process_options(opt, arg)
|
||||
@@ -80,6 +84,8 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
self.employees << JSON.parse(arg);
|
||||
when '--product_name'
|
||||
self.product_name << arg;
|
||||
when '--intro_paragraph'
|
||||
self.intro_paragraph << arg;
|
||||
end
|
||||
end
|
||||
|
||||
@@ -93,7 +99,8 @@ class RandomOrganisationGenerator < StringEncoder
|
||||
'industry: ' + self.industry.to_s + print_string_padding +
|
||||
'manager: ' + self.manager.to_s + print_string_padding +
|
||||
'employees: ' + self.employees.to_s + print_string_padding +
|
||||
'product_name: ' + self.product_name.to_s
|
||||
'product_name: ' + self.product_name.to_s +
|
||||
'intro_paragraph: ' + self.intro_paragraph.to_s
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
@@ -25,6 +25,7 @@
|
||||
<read_fact>manager</read_fact>
|
||||
<read_fact>employees</read_fact>
|
||||
<read_fact>product_name</read_fact>
|
||||
<read_fact>intro_paragraph</read_fact>
|
||||
|
||||
<default_input into="business_name">
|
||||
<generator type="business_name"/>
|
||||
@@ -58,6 +59,9 @@
|
||||
<default_input into="product_name">
|
||||
<generator type="product"/>
|
||||
</default_input>
|
||||
<default_input into="intro_paragraph">
|
||||
<generator type="paragraph_generator"/>
|
||||
</default_input>
|
||||
|
||||
<output_type>account</output_type>
|
||||
</generator>
|
||||
|
||||
@@ -12,6 +12,7 @@ class RealOrganisationGenerator < StringEncoder
|
||||
attr_accessor :manager
|
||||
attr_accessor :employees
|
||||
attr_accessor :product_name
|
||||
attr_accessor :intro_paragraph
|
||||
attr_accessor :filler_char
|
||||
|
||||
def initialize
|
||||
@@ -27,6 +28,7 @@ class RealOrganisationGenerator < StringEncoder
|
||||
self.manager = {}
|
||||
self.employees = []
|
||||
self.product_name = ''
|
||||
self.intro_paragraph = ''
|
||||
self.filler_char = %w(_ -).sample
|
||||
end
|
||||
|
||||
@@ -59,6 +61,7 @@ class RealOrganisationGenerator < StringEncoder
|
||||
organisation_hash['manager'] = self.manager
|
||||
organisation_hash['employees'] = self.employees
|
||||
organisation_hash['product_name'] = self.product_name
|
||||
organisation_hash['intro_paragraph'] = self.intro_paragraph
|
||||
|
||||
self.outputs << organisation_hash.to_json
|
||||
end
|
||||
@@ -88,7 +91,8 @@ class RealOrganisationGenerator < StringEncoder
|
||||
['--industry', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--manager', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--employees', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--product_name', GetoptLong::REQUIRED_ARGUMENT]]
|
||||
['--product_name', GetoptLong::REQUIRED_ARGUMENT],
|
||||
['--intro_paragraph', GetoptLong::REQUIRED_ARGUMENT]]
|
||||
end
|
||||
|
||||
def process_options(opt, arg)
|
||||
@@ -114,6 +118,8 @@ class RealOrganisationGenerator < StringEncoder
|
||||
self.employees << JSON.parse(arg);
|
||||
when '--product_name'
|
||||
self.product_name << arg;
|
||||
when '--intro_paragraph'
|
||||
self.intro_paragraph << arg;
|
||||
end
|
||||
end
|
||||
|
||||
@@ -127,7 +133,8 @@ class RealOrganisationGenerator < StringEncoder
|
||||
'industry: ' + self.industry.to_s + print_string_padding +
|
||||
'manager: ' + self.manager.to_s + print_string_padding +
|
||||
'employees: ' + self.employees.to_s + print_string_padding +
|
||||
'product_name: ' + self.product_name.to_s
|
||||
'product_name: ' + self.product_name.to_s +
|
||||
'intro_paragraph: ' + self.intro_paragraph.to_s
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
@@ -27,6 +27,7 @@
|
||||
<read_fact>manager</read_fact>
|
||||
<read_fact>employees</read_fact>
|
||||
<read_fact>product_name</read_fact>
|
||||
<read_fact>intro_paragraph</read_fact>
|
||||
|
||||
<default_input into="business_name">
|
||||
<generator type="business_name"/>
|
||||
@@ -60,6 +61,10 @@
|
||||
<default_input into="product_name">
|
||||
<generator type="product"/>
|
||||
</default_input>
|
||||
<default_input into="intro_paragraph">
|
||||
<generator type="paragraph_generator"/>
|
||||
</default_input>
|
||||
|
||||
|
||||
<output_type>account</output_type>
|
||||
</generator>
|
||||
|
||||
Reference in New Issue
Block a user