|
a |
|
b/scripts/purge_project.py |
|
|
1 |
import os
|
|
|
2 |
import sys
|
|
|
3 |
import json
|
|
|
4 |
import logging
|
|
|
5 |
|
|
|
6 |
from pylons import g
|
|
|
7 |
|
|
|
8 |
from ming.orm import state, session
|
|
|
9 |
|
|
|
10 |
from pymongo.json_util import default
|
|
|
11 |
from pyforge import model as M
|
|
|
12 |
|
|
|
13 |
log = logging.getLogger(__name__)
|
|
|
14 |
|
|
|
15 |
MONGO_HOME=os.environ.get('MONGO_HOME', '')
|
|
|
16 |
MONGO_DUMP=os.path.join(MONGO_HOME, 'bin/mongodump')
|
|
|
17 |
MONGO_RESTORE=os.path.join(MONGO_HOME, 'bin/mongorestore')
|
|
|
18 |
|
|
|
19 |
def main():
|
|
|
20 |
if len(sys.argv) != 2:
|
|
|
21 |
log.error('Usage: %s <shortname>', sys.argv[0])
|
|
|
22 |
return
|
|
|
23 |
pname = sys.argv[1]
|
|
|
24 |
log.info('Purging %s', pname)
|
|
|
25 |
project = M.Project.query.get(shortname=pname)
|
|
|
26 |
if project is None:
|
|
|
27 |
log.fatal('Project %s not found', pname)
|
|
|
28 |
return
|
|
|
29 |
dump_project(project)
|
|
|
30 |
|
|
|
31 |
def dump_project(project):
|
|
|
32 |
pname = project.shortname
|
|
|
33 |
gid = project.tool_data.get('sfx', {}).get('group_id', project._id)
|
|
|
34 |
dirname = '%s-%s.purge' % (pname, gid)
|
|
|
35 |
log.info('Purging %s to %s', pname, dirname)
|
|
|
36 |
os.system('%s --db %s -o %s' % (
|
|
|
37 |
MONGO_DUMP, project.database, dirname))
|
|
|
38 |
with open(os.path.join(dirname, 'project.json'), 'w') as fp:
|
|
|
39 |
json.dump(state(project).document, fp, default=default)
|
|
|
40 |
project.shortname = 'deleted-%s' % gid
|
|
|
41 |
project.deleted = True
|
|
|
42 |
g.solr.delete(q='project_id_s:%s' % project._id)
|
|
|
43 |
session(project).flush()
|
|
|
44 |
|
|
|
45 |
if __name__ == '__main__':
|
|
|
46 |
main()
|