f dddlZddlZddlmZddlmZddlmZddl m Z ddl m Z GddZ y) N)Path)apps)settings)Seeder) AppliedSeedercheZdZdZgZedZedZeddZeddZ edZ y) SeederRegistryz The `SeederRegistry` class apply registered seeders when the server is run. seeder registering is doing by: @SeederRegistry.register as or SeederRegistry.register() as ct|ts td|jD]*}|j |j k(s*y|jj |y)zk Method and decorator to register the seeder-class in the seeders list to be seeded when the server is run zNOnly subclasses of Seeder class can be registered with SeederRegistry.registerN) issubclassr TypeErrorseeders_get_idappend)clsseeder cur_seeders r/var/www/teaching_timetable/msuas_timetable/my_venv/lib/python3.12/site-packages/django_seeding/seeder_registry.pyregisterzSeederRegistry.registersb&&)lm m++ J!!#vx'7'7'99  68$c~tjD]}|j}d}|dz}t|j|z }|j s?t jj|d|t|}t jj|}|jj|y)zp Method that import all `seeders.py` files in the installed apps to register them in the `SeederRegistry` class r z.py.N) rget_app_configsnamerpathexists importlibutilspec_from_file_locationstrmodule_from_specloader exec_module)r app_configapp_name module_name file_name file_pathspecmodules r import_allzSeederRegistry.import_all!s..0 0J!H#K#e+IZ__- 9I! ~~== !K=>Y[^_h[ij"88> ''/ 0rNc|j}ttfd|}tjj|Dcgc]}|j c}j t|k7rd}d}t|dz|z|jd|D]}|j| ycc}w) z Method that call seed methods for all registered seeders sort the seeders depending on the `priority` (less is applied earlier) Nc(|jvSN)r)ridss rz)SeederRegistry.seed_all..7s1AS1Hr)id__inzzzRunning Seeders: c"|jSr-) _get_priority)rs rr/z)SeederRegistry.seed_all..>s(<(<(>r)key)debug) r listfilterrobjectsrcountlenprintsort_seed)rr4r.r r BLUE_COLOR WHITE_COLORs ` rseed_allzSeederRegistry.seed_all.s++ ?6"H'RSG  ' 'w/WV0@/W ' X ^ ^ `dghodp p#J#K *22[@ A > ? &F LLuL % & 0XsCcJ|j|j||y)a Note: the decorator `@SeederRegistry.register` will be applied when the file is imported so, if the seeder class is written in another file (not in `seeders.py`) then it will not be imported then it will not be applied when the server is run so, to solve this problem you can import any file contains seeder inside the `AppConfig` class of your app )r4r.N)r*r?)rr4r.s rimport_all_then_seed_allz'SeederRegistry.import_all_then_seed_allBs   5c *rcdtjvrytttdd}dtjvrd}dtjvrd}|r|j yy)N runserverSEEDING_ON_RUNSERVERFz--seedTz --dont-seed)sysargvboolgetattrrrA)rseeds ron_runzSeederRegistry.on_runTs[ chh & GH&rWs&  )/X+X+r