forked from adamm/git-importer
27 lines
675 B
Python
27 lines
675 B
Python
import sys
|
|
|
|
import yaml
|
|
|
|
from lib.db import DB
|
|
from lib.db_revision import DBRevision
|
|
|
|
|
|
class TestExporter:
|
|
""" "Helper class to export data from production DB for tests"""
|
|
|
|
def __init__(self, package):
|
|
self.package = package
|
|
|
|
def run(self):
|
|
db = DB()
|
|
with db.cursor() as cur:
|
|
cur.execute(
|
|
"SELECT * from revisions where package=%s ORDER BY project,rev",
|
|
(self.package,),
|
|
)
|
|
data = {"revisions": []}
|
|
for row in cur.fetchall():
|
|
data["revisions"].append(DBRevision(row).as_dict(db))
|
|
|
|
yaml.dump(data, sys.stdout, default_flow_style=False)
|