You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
3.2 KiB
69 lines
3.2 KiB
import pydash as _;
|
|
from django.core.management import BaseCommand
|
|
|
|
_.map = _.map_;
|
|
_.filter = _.filter_
|
|
|
|
from archilance import util
|
|
from common.models import Location
|
|
from projects.models import Portfolio, CURRENCIES, TERM_TYPES, BuildingClassfication, ConstructionType
|
|
from specializations.models import Specialization
|
|
from users.models import User
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
print('---------------------------------------')
|
|
print('Generating portfolios...')
|
|
print('---------------------------------------')
|
|
|
|
# ('photos', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('budget', 'Relation? False', 'Null? True', 'Blank? True', 'Hidden? False'),
|
|
# ('building_classification', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('construction_type', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('currency', 'Relation? False', 'Null? True', 'Blank? True', 'Hidden? False'),
|
|
# ('description', 'Relation? False', 'Null? False', 'Blank? False', 'Hidden? False'),
|
|
# ('location', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('name', 'Relation? False', 'Null? False', 'Blank? False', 'Hidden? False'),
|
|
# ('specialization', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('team', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('term', 'Relation? False', 'Null? True', 'Blank? True', 'Hidden? False'),
|
|
# ('term_type', 'Relation? False', 'Null? True', 'Blank? True', 'Hidden? False'),
|
|
# ('user', 'Relation? True', 'Null? True', '(relation)', 'Hidden? False'),
|
|
# ('worksell', 'Relation? False', 'Null? False', 'Blank? True', 'Hidden? False'),
|
|
|
|
|
|
def create_portfolio(i):
|
|
portf = Portfolio(
|
|
budget=util.random_amount(),
|
|
created=util.random_date(),
|
|
currency=_.sample(CURRENCIES)[0],
|
|
description=util.lorem(),
|
|
name=util.lorem(words=_.random(2, 20), sentences=1),
|
|
term=_.random(0, 20),
|
|
term_type=_.sample(TERM_TYPES)[0],
|
|
worksell=_.sample((True, False)),
|
|
)
|
|
|
|
portf.save()
|
|
|
|
portf.user = User.contractor_objects.order_by('?').first()
|
|
|
|
portf.building_classification = BuildingClassfication.objects.order_by('?').first()
|
|
portf.construction_type = ConstructionType.objects.order_by('?').first()
|
|
portf.location = Location.objects.root_nodes()[0].get_descendants().order_by('?').first()
|
|
portf.specialization = Specialization.objects.root_nodes()[0].get_descendants().order_by('?').first()
|
|
|
|
pic_names = _.split(
|
|
'a.png b.png c.jpg d.png e.jpg f.png g.png h.jpg i.png j.png k.jpg l.png m.png n.png o.png p.png q.jpg',
|
|
' ')
|
|
|
|
_.times(
|
|
lambda i_: portf.photos.create(img='_sample_files/%s' % util.take_one_random(pic_names)),
|
|
_.random(1, 15),
|
|
)
|
|
|
|
portf.save()
|
|
return portf
|
|
|
|
_.times(create_portfolio, 1000)
|
|
|