@@ -157,13 +157,13 @@ def create_models(type=None, integration=False):
157
157
obj_ids = []
158
158
with transaction .atomic ():
159
159
map_data , user_data , people_data , dataset_data , document_data = create_fixtures ()
160
- registeredmembers_group , created = Group .objects .get_or_create (name = 'registered-members' )
161
- anonymous_group , created = Group .objects .get_or_create (name = 'anonymous' )
162
- cont_group , created = Group .objects .get_or_create (name = 'contributors' )
160
+ registeredmembers_group , _ = Group .objects .get_or_create (name = 'registered-members' )
161
+ anonymous_group , _ = Group .objects .get_or_create (name = 'anonymous' )
162
+ cont_group , _ = Group .objects .get_or_create (name = 'contributors' )
163
163
perm = Permission .objects .get (codename = 'add_resourcebase' )
164
164
cont_group .permissions .add (perm )
165
165
logger .debug ("[SetUp] Get or create user admin" )
166
- u , created = get_user_model ().objects .get_or_create (username = 'admin' )
166
+ u , _ = get_user_model ().objects .get_or_create (username = 'admin' )
167
167
u .set_password ('admin' )
168
168
u .is_superuser = True
169
169
u .first_name = 'admin'
@@ -174,7 +174,7 @@ def create_models(type=None, integration=False):
174
174
for ud , pd in zip (user_data , cycle (people_data )):
175
175
user_name , password , first_name , last_name = ud
176
176
logger .debug (f"[SetUp] Get or create user { user_name } " )
177
- u , created = get_user_model ().objects .get_or_create (username = user_name )
177
+ u , _ = get_user_model ().objects .get_or_create (username = user_name )
178
178
u .set_password (password )
179
179
u .first_name = first_name
180
180
u .last_name = last_name
@@ -194,18 +194,19 @@ def create_models(type=None, integration=False):
194
194
for md , user in zip (map_data , cycle (users )):
195
195
title , abstract , kws , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), category = md
196
196
logger .debug (f"[SetUp] Add map { title } " )
197
- m = Map (
198
- uuid = str (uuid4 ()),
197
+ m , _ = Map .objects .get_or_create (
199
198
title = title ,
200
- abstract = abstract ,
201
- owner = user ,
202
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
203
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
204
- srid = 'EPSG:4326' ,
205
- category = category ,
206
- metadata_only = title == 'map metadata true'
199
+ defaults = dict (
200
+ uuid = str (uuid4 ()),
201
+ abstract = abstract ,
202
+ owner = user ,
203
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
204
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
205
+ srid = 'EPSG:4326' ,
206
+ category = category ,
207
+ metadata_only = title == 'map metadata true'
208
+ )
207
209
)
208
- m .save ()
209
210
m .set_default_permissions (owner = user )
210
211
m .clear_dirty_state ()
211
212
m .set_processing_state (enumerations .STATE_PROCESSED )
@@ -218,19 +219,20 @@ def create_models(type=None, integration=False):
218
219
for dd , user in zip (document_data , cycle (users )):
219
220
title , abstract , kws , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), category = dd
220
221
logger .debug (f"[SetUp] Add document { title } " )
221
- m = Document (
222
- uuid = str (uuid4 ()),
222
+ m , _ = Document .objects .get_or_create (
223
223
title = title ,
224
- abstract = abstract ,
225
- owner = user ,
226
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
227
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
228
- srid = 'EPSG:4326' ,
229
- files = dfile ,
230
- extension = "gif" ,
231
- metadata_only = title == 'doc metadata true'
224
+ defaults = dict (
225
+ uuid = str (uuid4 ()),
226
+ abstract = abstract ,
227
+ owner = user ,
228
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
229
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
230
+ srid = 'EPSG:4326' ,
231
+ files = dfile ,
232
+ extension = "gif" ,
233
+ metadata_only = title == 'doc metadata true'
234
+ )
232
235
)
233
- m .save ()
234
236
m .set_default_permissions (owner = user )
235
237
m .clear_dirty_state ()
236
238
m .set_processing_state (enumerations .STATE_PROCESSED )
@@ -244,24 +246,25 @@ def create_models(type=None, integration=False):
244
246
title , abstract , name , alternate , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), start , kws , category = ld
245
247
end = start + timedelta (days = 365 )
246
248
logger .debug (f"[SetUp] Add dataset { title } " )
247
- dataset = Dataset (
248
- title = title ,
249
- abstract = abstract ,
250
- name = name ,
249
+ dataset , _ = Dataset .objects .get_or_create (
251
250
alternate = alternate ,
252
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
253
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
254
- srid = 'EPSG:4326' ,
255
- uuid = str (uuid4 ()),
256
- owner = user ,
257
- temporal_extent_start = start ,
258
- temporal_extent_end = end ,
259
- date = start ,
260
- subtype = subtype ,
261
- category = category ,
262
- metadata_only = title == 'dataset metadata true'
251
+ defaults = dict (
252
+ title = title ,
253
+ abstract = abstract ,
254
+ name = name ,
255
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
256
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
257
+ srid = 'EPSG:4326' ,
258
+ uuid = str (uuid4 ()),
259
+ owner = user ,
260
+ temporal_extent_start = start ,
261
+ temporal_extent_end = end ,
262
+ date = start ,
263
+ subtype = subtype ,
264
+ category = category ,
265
+ metadata_only = title == 'dataset metadata true'
266
+ )
263
267
)
264
- dataset .save ()
265
268
dataset .set_default_permissions (owner = user )
266
269
dataset .clear_dirty_state ()
267
270
dataset .set_processing_state (enumerations .STATE_PROCESSED )
@@ -332,26 +335,27 @@ def create_single_dataset(name, keywords=None, owner=None, group=None, **kwargs)
332
335
ll = (name , 'lorem ipsum' , name , f'geonode:{ name } ' , [
333
336
0 , 22 , 0 , 22 ], test_datetime , ('populartag' ,), "farming" )
334
337
title , abstract , name , alternate , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), start , kws , category = ll
335
- dataset = Dataset (
336
- title = title ,
337
- abstract = abstract ,
338
- name = name ,
338
+ dataset , _ = Dataset .objects .get_or_create (
339
339
alternate = alternate ,
340
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
341
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
342
- srid = 'EPSG:4326' ,
343
- uuid = str (uuid4 ()),
344
- owner = owner or admin ,
345
- temporal_extent_start = test_datetime ,
346
- temporal_extent_end = test_datetime ,
347
- date = start ,
348
- subtype = "vector" ,
349
- resource_type = "dataset" ,
350
- typename = f"geonode:{ title } " ,
351
- group = group ,
352
- ** kwargs
340
+ defaults = dict (
341
+ title = title ,
342
+ abstract = abstract ,
343
+ name = name ,
344
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
345
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
346
+ srid = 'EPSG:4326' ,
347
+ uuid = str (uuid4 ()),
348
+ owner = owner or admin ,
349
+ temporal_extent_start = test_datetime ,
350
+ temporal_extent_end = test_datetime ,
351
+ date = start ,
352
+ subtype = "vector" ,
353
+ resource_type = "dataset" ,
354
+ typename = f"geonode:{ title } " ,
355
+ group = group ,
356
+ ** kwargs
357
+ )
353
358
)
354
- dataset .save ()
355
359
356
360
if isinstance (keywords , list ):
357
361
dataset = add_keywords_to_resource (dataset , keywords )
@@ -362,7 +366,7 @@ def create_single_dataset(name, keywords=None, owner=None, group=None, **kwargs)
362
366
return dataset
363
367
364
368
365
- def create_single_map (name , owner = None ):
369
+ def create_single_map (name , owner = None , ** kwargs ):
366
370
admin , created = get_user_model ().objects .get_or_create (username = 'admin' )
367
371
if created :
368
372
admin .is_superuser = True
@@ -373,24 +377,26 @@ def create_single_map(name, owner=None):
373
377
ll = (name , 'lorem ipsum' , name , f'{ name } ' , [
374
378
0 , 22 , 0 , 22 ], test_datetime , ('populartag' ,))
375
379
title , abstract , name , alternate , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), start , kws = ll
376
- m = Map (
377
- uuid = str (uuid4 ()),
380
+ m , _ = Map .objects .get_or_create (
378
381
title = title ,
379
- abstract = abstract ,
380
- owner = owner or admin ,
381
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
382
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
383
- srid = 'EPSG:4326' ,
384
- resource_type = "map"
382
+ defaults = dict (
383
+ uuid = str (uuid4 ()),
384
+ abstract = abstract ,
385
+ owner = owner or admin ,
386
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
387
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
388
+ srid = 'EPSG:4326' ,
389
+ resource_type = "map" ,
390
+ ** kwargs
391
+ )
385
392
)
386
- m .save ()
387
393
m .set_default_permissions (owner = owner or admin )
388
394
m .clear_dirty_state ()
389
395
m .set_processing_state (enumerations .STATE_PROCESSED )
390
396
return m
391
397
392
398
393
- def create_single_doc (name , owner = None ):
399
+ def create_single_doc (name , owner = None , ** kwargs ):
394
400
admin , created = get_user_model ().objects .get_or_create (username = 'admin' )
395
401
if created :
396
402
admin .is_superuser = True
@@ -402,18 +408,20 @@ def create_single_doc(name, owner=None):
402
408
0 , 22 , 0 , 22 ], test_datetime , ('populartag' ,))
403
409
title , abstract , name , alternate , (bbox_x0 , bbox_x1 , bbox_y0 , bbox_y1 ), start , kws = dd
404
410
logger .debug (f"[SetUp] Add document { title } " )
405
- m = Document (
406
- uuid = str (uuid4 ()),
411
+ m , _ = Document .objects .get_or_create (
407
412
title = title ,
408
- abstract = abstract ,
409
- owner = owner or admin ,
410
- bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
411
- ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
412
- srid = 'EPSG:4326' ,
413
- files = dfile ,
414
- resource_type = "document"
413
+ defaults = dict (
414
+ uuid = str (uuid4 ()),
415
+ abstract = abstract ,
416
+ owner = owner or admin ,
417
+ bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
418
+ ll_bbox_polygon = Polygon .from_bbox ((bbox_x0 , bbox_y0 , bbox_x1 , bbox_y1 )),
419
+ srid = 'EPSG:4326' ,
420
+ files = dfile ,
421
+ resource_type = "document" ,
422
+ ** kwargs
423
+ )
415
424
)
416
- m .save ()
417
425
m .set_default_permissions (owner = owner or admin )
418
426
m .clear_dirty_state ()
419
427
m .set_processing_state (enumerations .STATE_PROCESSED )
@@ -423,7 +431,6 @@ def create_single_doc(name, owner=None):
423
431
def add_keywords_to_resource (resource , keywords ):
424
432
for keyword in keywords :
425
433
resource .keywords .add (keyword )
426
-
427
434
resource .save ()
428
435
return resource
429
436
0 commit comments