From 1715f5dffed575e8bf4b8bac8bf881527e27cc76 Mon Sep 17 00:00:00 2001 From: Francesco Siddi Date: Tue, 7 Apr 2015 14:40:39 +0200 Subject: [PATCH] Update to manage.py and rename config example --- ...config.py.example.py => config.py.example} | 0 attract/init_bd.py | 109 ---------------- attract/manage.py | 122 +++++++++++++++++- 3 files changed, 116 insertions(+), 115 deletions(-) rename attract/{config.py.example.py => config.py.example} (100%) delete mode 100644 attract/init_bd.py diff --git a/attract/config.py.example.py b/attract/config.py.example similarity index 100% rename from attract/config.py.example.py rename to attract/config.py.example diff --git a/attract/init_bd.py b/attract/init_bd.py deleted file mode 100644 index f69307f8..00000000 --- a/attract/init_bd.py +++ /dev/null @@ -1,109 +0,0 @@ -from pymongo import MongoClient -from pymongo.errors import DuplicateKeyError -from bson import ObjectId -from datetime import datetime - -client = MongoClient() -db = client.eve - -default_user = { - "_id": ObjectId("550171c8135d3248e477f288"), - "_updated": datetime.now(), - "firstname": "admin", - "lastname": "admin", - "role": "admin", - "email": "admin@admin.com", - "_created": datetime.now(), - "_etag": "302236e27f51d2e26041ae9de49505d77332b260" - } - -default_token = { - "_id": ObjectId("5502f289135d3274cb658ba7"), - "username": "admin", - "token": "ANLGNSIEZJ", - "_etag": "1e96ed46b133b7ede5ce6ef0d6d4fc53edd9f2ba" - } - -shot_node_type = { - "_id": ObjectId("55016a52135d32466fc800be"), - "_updated": datetime.now(), - "name": "shot", - "description": "Shot Node Type, for shots", - "dyn_schema": { - "url": { - "type": "string", - }, - "cut_in": { - "type": "integer" - }, - "cut_out": { - "type": "integer" - }, - "status": { - "type": "string", - "allowed": ["on_hold", - "todo", - "in_progress", - "review_required", - "final"], - }, - "notes": { - "type": "string", - "maxlength": 256, - }, - "order": { - "type": "integer", - }, - "shot_group": { - "type": "string", - #"data_relation": { - # "resource": "nodes", - # "field": "_id", - #}, - } - }, - "_created": datetime.now(), - "_etag": "0ea3c4f684a0cda85525184d5606c4f4ce6ac5f5" - } - -shot = { - "_id": ObjectId("55016a52135d32466fc800be"), - "_update": datetime.now(), - "_created": datetime.now(), - "_etag": "0ea3c4f684a0cda85525184d5606c4f4ce6ac5f5", - "name": "01", - "description": "A sheep tries to hang itself, but fails", - "thumbnail": "/tmp/attrackt-thumbnail.png", - "order": 0, - "parent": None, - "node_type": "55016a52135d32466fc800be", - "properties": { - "url": "shot01", - "cut_in": 100, - "cut_out": 900, - "status": "todo", - "notes": "I think the sheep should scream a bit more", - "order": 1, - "shot_group": "", - } -} - -try: - db.users.insert(default_user) -except DuplicateKeyError: - print ("default_user already exist") - -try: - db.node_types.insert(shot_node_type) -except DuplicateKeyError: - print ("shot_node_type already exist") - -try: - db.tokens.insert(default_token) -except DuplicateKeyError: - print ("default_token already exist") - -try: - db.nodes.insert(shot) -except DuplicateKeyError: - print ("shot already exist") diff --git a/attract/manage.py b/attract/manage.py index de8848cc..e6fc23c7 100644 --- a/attract/manage.py +++ b/attract/manage.py @@ -3,12 +3,6 @@ from flask.ext.script import Manager manager = Manager(app) - -@manager.command -def create_all_tables(): - pass - #db.create_all() - @manager.command def runserver(): try: @@ -26,5 +20,121 @@ def runserver(): host=HOST, debug=DEBUG) + +@manager.command +def populate_db_test(): + """Populate the db with sample data + """ + from pymongo import MongoClient + from pymongo.errors import DuplicateKeyError + from bson import ObjectId + from datetime import datetime + + client = MongoClient() + db = client.eve + + default_user = { + "_id": ObjectId("550171c8135d3248e477f288"), + "_updated": datetime.now(), + "firstname": "admin", + "lastname": "admin", + "role": "admin", + "email": "admin@admin.com", + "_created": datetime.now(), + "_etag": "302236e27f51d2e26041ae9de49505d77332b260" + } + + default_token = { + "_id": ObjectId("5502f289135d3274cb658ba7"), + "username": "admin", + "token": "ANLGNSIEZJ", + "_etag": "1e96ed46b133b7ede5ce6ef0d6d4fc53edd9f2ba" + } + + shot_node_type = { + "_id": ObjectId("55016a52135d32466fc800be"), + "_updated": datetime.now(), + "name": "shot", + "description": "Shot Node Type, for shots", + "dyn_schema": { + "url": { + "type": "string", + }, + "cut_in": { + "type": "integer" + }, + "cut_out": { + "type": "integer" + }, + "status": { + "type": "string", + "allowed": ["on_hold", + "todo", + "in_progress", + "review_required", + "final"], + }, + "notes": { + "type": "string", + "maxlength": 256, + }, + "order": { + "type": "integer", + }, + "shot_group": { + "type": "string", + #"data_relation": { + # "resource": "nodes", + # "field": "_id", + #}, + } + }, + "_created": datetime.now(), + "_etag": "0ea3c4f684a0cda85525184d5606c4f4ce6ac5f5" + } + + shot = { + "_id": ObjectId("55016a52135d32466fc800be"), + "_update": datetime.now(), + "_created": datetime.now(), + "_etag": "0ea3c4f684a0cda85525184d5606c4f4ce6ac5f5", + "name": "01", + "description": "A sheep tries to hang itself, but fails", + "thumbnail": "/tmp/attrackt-thumbnail.png", + "order": 0, + "parent": None, + "node_type": "55016a52135d32466fc800be", + "properties": { + "url": "shot01", + "cut_in": 100, + "cut_out": 900, + "status": "todo", + "notes": "I think the sheep should scream a bit more", + "order": 1, + "shot_group": "", + } + } + + try: + db.users.insert(default_user) + except DuplicateKeyError: + print ("default_user already exist") + + try: + db.node_types.insert(shot_node_type) + except DuplicateKeyError: + print ("shot_node_type already exist") + + try: + db.tokens.insert(default_token) + except DuplicateKeyError: + print ("default_token already exist") + + try: + db.nodes.insert(shot) + except DuplicateKeyError: + print ("shot already exist") + + if __name__ == '__main__': manager.run()