def test_pick_object_fields_to_field_not_have_in_obj(self):
class Person:
name = "tung"
age = 23
country = "QN"
self.assertEqual(pick_object_fields(Person,[]),Person)
self.assertEqual(pick_object_fields(Person,["names"]), {})
self.assertEqual(pick_object_fields(Person, ["names","age","country"]),dict(age=23,country="QN"))
def test_pick_object_fields_ok(self):
class Person:
name = "tung"
age = 23
country = "QN"
_field_names = ["name","age","country"]
self.assertEqual(pick_object_fields(Person,_field_names),dict(name="tung",age=23,country="QN"))
def test_pick_object_fields_to_obj_is_undefined(self):
self.assertEqual(pick_object_fields(None,[]),None)
self.assertEqual(pick_object_fields(None,["name"]),{})
def test_pick_object_fields_to_obj_is_dict(self):
_obj = {"name":"tung","age":23,"country":"QN"}
self.assertEqual(pick_object_fields(_obj,["names"]),{})
self.assertEqual(pick_object_fields(_obj,["name"]),dict(name="tung"))
self.assertEqual(pick_object_fields(_obj,["name","age","country"]),dict(name="tung",age=23,country="QN"))
def main(argv=sys.argv):
parser = argparse.ArgumentParser(prog='python3 -m modelmock')
subparsers = parser.add_subparsers(help='Sub-commands', dest='sub_command')
# create the parser for the "generate" command
generate_abc_parser = subparsers.add_parser(
'generate', help='Generate a type-based collection of records')
generate_subparsers = generate_abc_parser.add_subparsers(
help='[generate] sub-commands', dest='generated_target')
generate_agents_parser = generate_subparsers.add_parser(
'agents', help='Generate a collection of agents')
generate_agents_parser.add_argument(
"--total", type=int, help="Number of agents will be generated")
generate_agents_parser.add_argument("--mappings",
type=str,
help="level_mappings in JSON format")
generate_agents_parser.add_argument("--id_method",
type=str,
choices=['incr', 'uuid'],
help="The ID generation method")
generate_agents_parser.add_argument("--id_prefix",
type=str,
default="AGENT",
help="Prefix of the Agent IDs")
generate_agents_parser.add_argument("--id_padding",
type=int,
default=4,
help="Length of padded numbers of IDs")
generate_agents_parser.add_argument("--id_shuffle",
type=str,
default="true",
choices=CHOICES_VALUE,
help="The IDs will be shuffled or not")
generate_agents_parser.add_argument(
"--locale",
type=str,
help=
"The locale which used to generate the user information (phone, email, name, ...)"
)
generate_candidates_parser = generate_subparsers.add_parser(
'candidates', help='Generate a collection of candidates')
generate_candidates_parser.add_argument(
"--total", type=int, help="Number of candidates will be generated")
generate_promocodes_parser = generate_subparsers.add_parser(
'promocodes', help='Generate a collection of promotion-codes')
generate_promocodes_parser.add_argument(
"--total",
type=int,
help="Number of promotion-codes will be generated")
generate_contracts_parser = generate_subparsers.add_parser(
'contracts', help='Generate a collection of contracts')
generate_contracts_parser.add_argument(
"--total",
type=int,
required=True,
help="Number of contracts will be generated")
generate_contracts_parser.add_argument(
"--price",
type=int,
required=True,
help="Average price of contracts will be generated")
generate_contracts_parser.add_argument(
"--multiplier",
type=int,
default=1,
help="Multiplier of contract price (1, 10, 100, 1000, ...)")
generate_contracts_parser.add_argument(
"--extra_descriptors",
type=str,
default="[]",
help="extra_descriptors in JSON format")
generate_purchases_parser = generate_subparsers.add_parser(
'purchases', help='Generate a collection of purchases')
generate_purchases_parser.add_argument(
"--total_agents",
type=int,
required=True,
help="Number of agents associated with purchases")
generate_purchases_parser.add_argument(
"--total_contracts",
type=int,
required=True,
help="Number of contracts associated with purchases")
generate_purchases_parser.add_argument(
"--contract_price",
type=int,
required=True,
help="Average price of purchases will be generated")
generate_purchases_parser.add_argument(
"--multiplier",
type=int,
default=1,
help="Multiplier of contract price (1, 10, 100, 1000, ...)")
generate_purchases_parser.add_argument(
"--extra_descriptors",
type=str,
default="[]",
help="extra_descriptors in JSON format")
generate_purchases_parser.add_argument("--agent_id_method",
type=str,
choices=['incr', 'uuid'],
help="The ID generation method")
generate_purchases_parser.add_argument("--agent_id_prefix",
type=str,
default="AGENT",
help="Prefix of the Agent IDs")
generate_purchases_parser.add_argument(
"--agent_id_padding",
type=int,
default=4,
help="Length of padded numbers of IDs")
generate_purchases_parser.add_argument(
"--agent_id_shuffle",
type=str,
default="true",
choices=CHOICES_VALUE,
help="The IDs will be shuffled or not")
generate_purchases_parser.add_argument("--contract_id_method",
type=str,
choices=['incr', 'uuid'],
help="The ID generation method")
generate_purchases_parser.add_argument("--contract_id_prefix",
type=str,
default="CONTR",
help="Prefix of the Contract IDs")
generate_purchases_parser.add_argument(
"--contract_id_padding",
type=int,
default=4,
help="Length of padded numbers of IDs")
generate_purchases_parser.add_argument(
"--contract_id_shuffle",
type=str,
default="true",
choices=CHOICES_VALUE,
help="The IDs will be shuffled or not")
args = parser.parse_args(args=argv[1:])
if args.sub_command == 'generate':
if args.generated_target == 'agents':
display(
AgentsFaker(**dict(
total_agents=args.total,
level_mappings=json.loads(args.mappings),
id_method=args.id_method,
id_prefix=args.id_prefix,
id_padding=args.id_padding,
id_shuffle=(
args.id_shuffle is not None
and args.id_shuffle.lower() in ['true', 'yes', '1']),
locale=args.locale,
)))
return 0
if args.generated_target == 'candidates':
display(CandidatesFaker(args.total))
return 0
if args.generated_target in [
'promocodes', 'promotion-codes', 'promotion_codes'
]:
display(PromocodesFaker(args.total))
return 0
if args.generated_target == 'contracts':
display(
ContractsFaker(**dict(
total_contracts=args.total,
contract_price=args.price,
multiplier=args.multiplier,
extra_descriptors=json.loads(args.extra_descriptors),
)))
return 0
if args.generated_target == 'purchases':
pprint(
list(
generate_purchases(
**pick_object_fields(args, [
'total_agents',
'total_contracts',
'contract_price',
'multiplier',
'extra_descriptors',
'agent_id_method',
'agent_id_prefix',
'agent_id_padding',
'agent_id_shuffle',
'contract_id_method',
'contract_id_prefix',
'contract_id_padding',
'contract_id_shuffle',
]))))
return 0
parser.print_help()
return -1