fJ:DddlZddlZddlmZddlmZddlmZddlm Z ddl m Z GddZ Gd d e Z Gd d e ZGd de ZGddeZGddZGddZGddeeZGddeeZGddeeZGddeeZy)N)models)settings) HttpRequest) serializers) AppliedSeedercHeZdZdZdZd defdZdZdZdZ d Z d Z d Z y) Seedera The `Seeder` class provides a minimal class which may be used for writing custom seeding implementations. Required: seed: `seed()` as Additionals: priority: `priority` as or `get_priority()` as just_debug: `just_debug` as or `get_just_debug()` as : ctd)z) Method that fill the datebase as wanted z`seed()` must be implemented.)NotImplementedErrorselfs j/var/www/teaching_timetable/msuas_timetable/my_venv/lib/python3.12/site-packages/django_seeding/seeders.pyseedz Seeder.seeds!"ABBNdebugc|tj}|jr|sy|j}tj j |jryd}d}d}t|d|d|d |jtj j|t|d |y) zK Inner method that do validation before calling the public `seed()` method N)idzzzz Seeding z...)endzSuccessfully ^_^) rDEBUG_get_just_debug_get_idrobjectsfilterexistsprintrcreate)r rr PURPLE_COLOR GREEN_COLOR WHITE_COLORs r_seedz Seeder._seed"s =NNE    !%  \\^  ' '2 ' . 5 5 7 !    jC }=2F  $$$+  -k]; str(type(self)) Note: if you changed the class name or changed the seeder-class file name or and file in the path from the root to the class the str(type(self)) will return another value then the default value of this seeder is changed then if it doesnt have a constant id the seeder will be applied again and it may cause errors so: give an `id` class attribute to solv this problem r)r%strtyper s rget_idz Seeder.get_idls.tT3tDz?33rc\|j}t|ts td|S)zC Innder method to validate the value returned by `get_id()` method z`id` must be str)r5r)r3r+)r rs rrzSeeder._get_ids( [[]"c"./ / rN) __name__ __module__ __qualname____doc__rr1r!r'r,r/rr5rrrr r s7&C=4=< 7 242rr ceZdZdZdZdZy) DataSeederz The `DataSeeder` class provides a minimal class which may be used for writing custom seeding implementations. Required: data: `data` as or `get_data()` as c:t|dd}| td|S)z. Method return the `data` that will be seeded dataNzVsubclasses of `DataSeeder` must have `data` class attribute or the `get_data()` methodr%r+)r r@s rget_datazDataSeeder.get_datas&tVT* <tu u rc|j}d}t|ts t||D]}t|trt||S)zE Innder method to validate the value returned by `get_data()` method z`data` must be list of dict)rBr)listr+dict)r r@ error_message record_datas r _get_datazDataSeeder._get_datasR}}5 $%M* * /Kk40 .. / rN)r8r9r:r;rBrHr<rrr>r>s  rr>c"eZdZdZdZdZdZy) ModelSeedera The `ModelSeeder` class is a subclasse of `DataSeeder` needs `model` and provides fast `seed()` implementation with `bulk_create()` method. Required: model: `model` as or `get_model()` as c:t|dd}| td|S)z/ Method return the `model` that will be seeded modelNzYsubclasses of `ModelSeeder` must have `model` class attribute or the `get_model()` methodrAr rLs r get_modelzModelSeeder.get_models&gt, =wx x rc|j}t|trt|tj s t d|S)zF Innder method to validate the value returned by `get_model()` method z7`model` must be a subclasse of `django.db.models.Model`)rNr)r4 issubclassrModelr+rMs r _get_modelzModelSeeder._get_models6 %&j .MUV V rc|j}|j}|Dcgc] }|di| }}|jj|ycc}w)zA Standard Implementation of `seed()` method with `bulk_create()` Nr<)rHrRr bulk_create)r r@rLrG new_objectss rrzModelSeeder.seedsM~~!?CD u+{+D D !!+.EsAN)r8r9r:r;rNrRrr<rrrJrJs/rrJc"eZdZdZdZdZdZy)SerializerSeedera The `SerializerSeeder` class is a subclasse of `DataSeeder` needs `serializer_class` and provides slow `seed()` implementation. Note: this class is slow not like `ModelSeeder`, so if you have a big dataset dont use this class Required: serializer_class: `serializer_class` as or `get_serializer_class()` as c:t|dd}| td|S)zC Method return the `serializer_class` that will be used in seeding serializer_classNzlsubclasses of SerializerSeeder must have serializer_class class attribute or the get_serializer_class methodrAr rYs rget_serializer_classz%SerializerSeeder.get_serializer_classs3"4);TB  #KL Lrc|j}t|trt|tj s t d|S)zQ Innder method to validate the value returned by `get_serializer_class()` method zMserializer_class must be a subclasse of rest_framework.serializers.Serializer)r[r)r4rPr Serializerr+rZs r_get_serializer_classz&SerializerSeeder._get_serializer_classs?446*D1DTVaVlVl9mkl lrc|j}|j}|D]A}t}d|_||d|i}|j d|j Cy)zY Slow Implementation of `seed()` method with `serializer.save()` method for every record Nrequest)r@contextT)raise_exception)rHr^ruseris_validsave)r r@rYrGr` serializers rrzSerializerSeeder.seedsg~~557 K!mGGL){YPWDXYJ     5 OO   rN)r8r9r:r;r[r^rr<rrrWrWs    rrWc"eZdZdZdZdZdZy) EmptySeedera6 The `EmptySeeder` class is a subclasse of `ModelSeeder` needs `records_count` and provides fast `seed()` implementation with `bulk_create()` method. Required: records_count: `records_count` as or `get_records_count()` as c:t|dd}| td|S)z@ Method return the `records_count` that will be used in seeding records_countNzasubclasses of EmptySeeder must have records_count class attribute or the get_records_count methodrAr rjs rget_records_countzEmptySeeder.get_records_count s.ot<  A Arc\|j}t|ts td|S)zN Innder method to validate the value returned by `get_records_count()` method zrecords_count must be int)rlr)r*r+rks r_get_records_countzEmptySeeder._get_records_counts,..0 --78 8rc|j}|jfdt|D}jj |y)zS Standard Implementation of `seed()` method for empty objects with `bulk_create()` c3,K|] } ywr7r<).0_rLs r z#EmptySeeder.seed..!s=1uw=sN)rnrRrangerrT)r rjrUrLs @rrzEmptySeeder.seeds>//1 !=m(<=  !!+.rN)r8r9r:r;rlrnrr<rrrhrhs/rrhc"eZdZdZdZdZdZy) CSVFileReaderz The `CSVFileReader` class needs `csv_file_path` and provides `get_data()` implementation. Required: csv_file_path: `csv_file_path` as or `get_csv_file_path()` as c:t|dd}| td|S)z[ Method return the `csv_file_path` that will be used in `get_date()` method implementation csv_file_pathNzksubclasses of `CSVFileReader` must have `csv_file_path` class attribute or the `get_csv_file_path()` methodrAr rxs rget_csv_file_pathzCSVFileReader.get_csv_file_path/s1ot<  JK Krc\|j}t|ts td|S)zN Innder method to validate the value returned by `get_csv_file_path()` method z`csv_file_path` must be str)rzr)r3r+rys r_get_csv_file_pathz CSVFileReader._get_csv_file_path8s,..0 --9: :rcg}|j}tj|}|jD]5\}}i}|j D] \}}|||< |j |7|S)zx Method (using pandas) read the csv_file that is specified by `csv_file_path` and return the `data` that will be seeded )r|pandasread_csviterrowsitemsappend) r r@rx pandas_filerrrowrGkeyvalues rrBzCSVFileReader.get_dataAsz//1 oom4 !**, %FAsK!iik ) U#( C  ) KK $  %  rN)r8r9r:r;rzr|rBr<rrrvrv%s rrvc"eZdZdZdZdZdZy)JSONFileReadera The `JSONFileReader` class needs `json_file_path` and provides `get_data()` implementation. Required: json_file_path: `json_file_path` as or `get_json_file_path()` as c:t|dd}| td|S)z\ Method return the `json_file_path` that will be used in `get_date()` method implementation json_file_pathNznsubclasses of `JSONFileReader` must have `json_file_path` class attribute or the `get_json_file_path()` methodrAr rs rget_json_file_pathz!JSONFileReader.get_json_file_pathXs2 '7>  !MN Nrc\|j}t|ts td|S)zO Innder method to validate the value returned by `get_json_file_path()` method z`json_file_path` must be str)rr)r3r+rs r_get_json_file_pathz"JSONFileReader._get_json_file_pathas,002.#.:; ;rc|j}t|dd5}tj|}|cdddS#1swYyxYw)zk Method read the json_file that is specified by `json_file_path` and return the `data` that will be seeded rzutf-8)encodingN)ropenjsonload)r rfiler@s rrBzJSONFileReader.get_datajsC113 .# 8 D99T?D   s AA N)r8r9r:r;rrrBr<rrrrNsrrceZdZdZy)CSVFileModelSeedera The `CSVFileModelSeeder` class is a Fast Full implemented class that needs `csv_file_path`, 'model` and provides `seed()` implementation. Required: csv_file_path: `csv_file_path` as or `get_csv_file_path()` as model: `model` as or `get_model()` as Nr8r9r:r;r<rrrrr   rrceZdZdZy)CSVFileSerializerSeedera The `CSVFileSerializerSeeder` class is a Slow Full implemented class that needs `csv_file_path`, 'serializer_class` and provides `seed()` implementation. Required: csv_file_path: `csv_file_path` as or `get_csv_file_path()` as serializer_class: `serializer_class` as or `get_serializer_class()` as Nrr<rrrrrrrceZdZdZy)JSONFileModelSeedera The `JSONFileModelSeeder` class is a Fast Full implemented class that needs `json_file_path`, 'model` and provides `seed()` implementation. Required: json_file_path: `json_file_path` as or `get_json_file_path()` as model: `model` as or `get_model()` as Nrr<rrrrrrrceZdZdZy)JSONFileSerializerSeedera The `JSONFileSerializerSeeder` class is a Slow Full implemented class that needs `json_file_path`, 'serializer_class` and provides `seed()` implementation. Required: json_file_path: `json_file_path` as or `get_json_file_path()` as serializer_class: `serializer_class` as or `get_serializer_class()` as Nrr<rrrrrrr)r~r django.dbr django.confr django.httprrest_frameworkrdjango_seeding.modelsrr r>rJrWrhrvrrrrrr<rrrs  #&/BBJ!!H!/*!/H'z'T!/+!/H&&R!!H   $ m-= $ .+ $ ~/? r