Scott Baker | c41914d | 2017-06-26 14:00:52 -0700 | [diff] [blame^] | 1 | def CreateOrFind(kind, args): |
| 2 | objs=ListAll(kind, args.copy()) |
| 3 | if objs: |
| 4 | id_name = {"ServiceProvider": "service_provider_id", |
| 5 | "ContentProvider": "content_provider_id", |
| 6 | "OriginServer": "origin_server_id", |
| 7 | "CDNPrefix": "cdn_prefix_id"} |
| 8 | print kind, "exists with args", args |
| 9 | return objs[0].get(id_name[kind]) |
| 10 | else: |
| 11 | print "create", kind, "with args", args |
| 12 | return Create(kind, args) |
| 13 | sp=CreateOrFind("ServiceProvider", {"account": "cord", "name": "cord", "enabled": True}) |
| 14 | cp=CreateOrFind("ContentProvider", {"account": "test", "name": "test", "enabled": True, "service_provider_id": sp}) |
| 15 | ors=CreateOrFind("OriginServer", {"url": "http://www.cs.arizona.edu", "content_provider_id": cp, "service_type": "HyperCache"}) |
| 16 | pre=CreateOrFind("CDNPrefix", {"service": "HyperCache", "enabled": True, "content_provider_id": cp, "cdn_prefix": "test.vicci.org", "default_origin_server": "http://www.cs.arizona.edu"}) |
| 17 | cp=CreateOrFind("ContentProvider", {"account": "onlab", "name": "onlab", "enabled": True, "service_provider_id": sp}) |
| 18 | ors=CreateOrFind("OriginServer", {"url": "http://onlab.vicci.org", "content_provider_id": cp, "service_type": "HyperCache"}) |
| 19 | pre=CreateOrFind("CDNPrefix", {"service": "HyperCache", "enabled": True, "content_provider_id": cp, "cdn_prefix": "onlab.vicci.org", "default_origin_server": "http://onlab.vicci.org"}) |