fV&ddlZddlZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZGddeZGd d eZy) N) ForeignKeyManyToManyField OneToOneField)SeederException) NameGuesserFieldTypeGuesser)IntegrityError) transactioncPeZdZdZedZedZedZddZdZ y) ModelSeederc.||_i|_i|_y)z) :param model: Generator N)modelfield_formattersmany_relations)selfrs f/var/www/teaching_timetable/msuas_timetable/my_venv/lib/python3.12/site-packages/django_seed/seeder.py__init__zModelSeeder.__init__ s " cfd}|S)Nc|vr9|r4tj|}jj|Sjsdj }t |yNpkField {} cannot be null)randomchoiceobjectsgetnullformatr)insertedrmessagefield related_models rfuncz(ModelSeeder.build_relation..funcsf(Xm-D]]8M#:;$,,00B077ZZ3::5A%g.. rr#r$r%s`` rbuild_relationzModelSeeder.build_relations / rcfd}|S)Nc$|vrc|r^tt|z }|rBtj|}j |j j |Sjsdj}t|yr) listsetrraddrrrr r)r!unusedrr"existingr#r$s rr%z,ModelSeeder.build_one_relation..func"s(Xm-Dc(="9:XEFv.BLL$(00444;;::3::5A%g..rr&)r#r$r/r%s``` rbuild_one_relationzModelSeeder.build_one_relation s / rcfd}|S)Nc |vr|rtdtt|dz dz}g}tt j d|D]E}t j |}|jjj|G|Sjsdj}t|tjdjgS)N rrz?Could not build many-to-many relationship for between {} and {})minroundlenrangerrandintrappendrrblankr rloggingwarn)r! max_relations return_list_rr"r#r$s rr%z-ModelSeeder.build_many_relation..func2s(Xm-D #Bc(=2I.JQ.N(ORS(S T  v~~a?@MA#]]8M+BCF&&}'<'<'@'@F'@'KLM#"[[3::5A%g.. U\\%  rr&r's`` rbuild_many_relationzModelSeeder.build_many_relation0s , rNc |si}t|}t|}|jjjD]}|j }|j r||vr!|jr|j||<Et|tr+t}|j||j|||<t|tr |j||j||<|js|j!|}|r|||<|j!|}|s|||<|jjj"D]5}|j%||j|j&|j <7|S)a< Gets the formatter methods for each field using the guessers or related object fields :param faker: Faker factory object :param formatters: this is 'customFieldFormatters' - optional dict with field as key and callable as value :type formatters: dict or None )rrr_metafieldsname primary_key get_default isinstancerr,r0r$rr(choices guess_format many_to_manyrBr) rfaker formatters name_guesserfield_type_guesserr# field_namer/ formatters rguess_field_formattersz"ModelSeeder.guess_field_formattersJsJ"5) -e4ZZ%%,,% EJ   Z'  ").):):)< :&%/5)-)@)@5..* :&%,)-)<)I)2 :&K% NZZ%%22 E.2.F.Fu**/D   +  rc d}d}|jjj|}||j|jj Dcic]\}}||||}}}|D]]} |jj j | }|js5t|| tsI|| d|j|| <_|jdi|} |jj D]2\}} | |} | s| D]} t| |j| 4| jScc}}w)zo Execute the stages entities to insert :param using: :param inserted_entities: c,t|r||S|SN)callable)r inserted_entitiess r format_fieldz)ModelSeeder.execute..format_fields/00Mrc|jjD]+}t|ddrd|_t|dds%d|_-y)Nauto_nowF auto_now_add)rDrEgetattrr[r\)rr#s rturn_off_auto_addz.ModelSeeder.execute..turn_off_auto_addsB++ /5*e4%*EN5.%8).E&  /r)usingNr&)rr db_managerritemsrD get_field max_lengthrIstrcreaterr]r-r) rr_rXrYr^managerr# field_format faker_data data_fieldobjr+ related_objs rexecutezModelSeeder.executesP   /**$$//e/<'--((,'<'<'B'B'D #| < .?@ @  % TJJJ$$..z:EJz*/Es$K)3J)?@R%BRBR)S :&  T gnn*z*..446 9KE4)*D#'9KC'++K89 9 vv ) sErV) __name__ __module__ __qualname__r staticmethodr(r0rBrSrlr&rrr r sN!    2;z*rr c,eZdZdZddZdifdZdZy)Seederc ||_g|_y)z) :param faker: Generator N)rMorders)rrMs rrzSeeder.__init__s  rNct|}|j|j||_|j||d}|j j |y)a Add an order for the generation of $number records for $entity. :param model: mixed A Django Model classname, or a faker.orm.django.EntitySeeder instance :type model: Model :param number: int The number of entities to seed :type number: integer :param customFieldFormatters: optional dict with field as key and callable as value :type customFieldFormatters: dict or None )rN)klassquantityentityN)r rSrMrrrtr;)rrnumbercustomFieldFormattersorders r add_entityzSeeder.add_entitysZE"!&!=!= JJ#8">"  [[  5!rc |s|j}i}t|jr|jjd}|d}|d}|d}t j dj ||||vrg||<|dz}d}d} |dkDrZ tj5|j||} ddd||j |dz }||k(rn |dz}|dkDrZ|dk(rtd |jd | ||k7r!td |d |d |jd| t|jr|S#1swYxYw#t$r } | } Yd} ~ d} ~ wwxYw)z Populate the database using all the Entity classes previously added. :param using A Django database connection name :rtype: A list of the inserted PKs rrwrvrxzCreating {} of {}Nr5z+Error: could not generate any instances of z Internal error: zWarning: could only generate z out of z instances of z, the rest errored with; )get_connectionr8rtpopr=debugr r atomicrlr;r rmprint) rr_rXr{ryrvrxattemptscompleted_count last_errorexecuted_entityerrs rrlzSeeder.executes '')E$++KKOOA&E:&F'NE8_F MM-44VUC D--+-!%(zHOJQ, %%++-S*0..@Q*RS&e,33OD#q(O #f,A !Q,$!#$'RSXSaSaRbbtuuA&BCC F*5o5FhvhVdejesesdtuNOYNZ[\Q$++T! 'SS &%!$J%s0E&E 9!E EE E+E&&E+c|jDcgc]}|d }}|s d}t|t|d}|jjScc}w)zO use the first connection available :rtype: Connection rvz5No classes found. Did you add entities to the Seeder?r)rtrr+r_db)rr{klassesr"rvs rrzSeeder.get_connectionsU 04{{;e5>;;MG!'* *W a }}   rs7GG2>*!f&fRf!Vf!r