def agencies(app, request):
with app.app_context():
db.session.add(
Agency(
id=1,
name='Department of Adventure',
domain='adventure.gov.au',
category='Commonwealth',
state='ACT',
whitelisted=True,
domains=[AgencyDomain(domain='adventure.gov.au',
active=True)]))
db.session.add(
Agency(id=2,
name='Department of Millenium Falcons',
domain='falcon.net.au',
category='Commonwealth',
state='NSW',
whitelisted=True,
domains=[AgencyDomain(domain='falcon.net.au',
active=True)]))
db.session.add(
Agency(id=3,
name='Jedi Temple',
domain='jedi.edu.au',
category='State',
state='NSW',
whitelisted=False,
domains=[AgencyDomain(domain='jedi.edu.au', active=True)]))
db.session.commit()
yield Agency.query.all()
def agencies(app, request):
with app.app_context():
db.session.add(
Agency(
id=1,
name='Digital Transformation Agency',
domain='digital.gov.au',
category='Commonwealth',
whitelisted=True,
domains=[AgencyDomain(domain='digital.gov.au', active=True)]))
db.session.add(
Agency(id=2,
name='Test Agency',
domain='test.gov.au',
category='Commonwealth',
whitelisted=True,
domains=[AgencyDomain(domain='test.gov.au', active=True)]))
db.session.add(
Agency(id=3,
name='Another Test Agency',
domain='asdf.com.au',
category='Commonwealth',
whitelisted=True,
domains=[AgencyDomain(domain='asdf.com.au', active=True)]))
db.session.commit()
yield Agency.query.all()
def agencies(app):
with app.app_context():
if not db.session.query(Agency).filter(Agency.id == 10).one_or_none():
db.session.add(
Agency(id=10,
name='Team Test Agency',
domain='teamtest.gov.au',
category='Commonwealth',
whitelisted=True,
domains=[
AgencyDomain(domain='teamtest.gov.au', active=True)
]))
db.session.add(
Agency(id=11,
name='Team Test Agency 2',
domain='teamtest2.gov.au',
category='Commonwealth',
whitelisted=True,
domains=[
AgencyDomain(domain='teamtest2.gov.au', active=True)
]))
db.session.commit()
yield Agency.query.all()
def agencies(client, app):
with app.app_context():
db.session.add(
Agency(id=1,
name="TEST",
domain="test.gov.au",
category="Commonwealth",
whitelisted=True,
state="ACT",
body_type="other",
reports=True,
must_join_team=True))
db.session.add(
Agency(id=2,
name="TEST 2",
domain="test2.gov.au",
category="Commonwealth",
whitelisted=True,
state="ACT",
body_type="other",
reports=True,
must_join_team=False))
db.session.commit()
yield db.session.query(Agency).all()
def post(self, agency_id):
if agency_id:
agency = Agency.query.get(agency_id)
else:
agency = Agency()
db.session.add(agency)
form = AgencyForm(request.form, agency, csrf_enabled=False)
if form.validate_on_submit():
form.populate_obj(agency)
db.session.commit()
flash('Agency updated.', 'success')
return redirect(url_for('agencies.agencies'))
return render_template('form.html', form=form, agency=agency)
def agencies(self, app):
with app.app_context():
db.session.add(
Agency(
id=1,
name='Department of Schmidt',
domain='ng.gov.au',
whitelisted=True,
body_type='cc',
reports=False
)
)
db.session.commit()
yield db.session.query(Agency).all()
def get_or_add_agency(self, domain):
from app.tasks import publish_tasks
domain = domain.lower()
agency = self.get_agency_by_domain(domain)
if not agency:
agency = self.save(
Agency(name=domain,
domain=domain,
category='Commonwealth',
state='ACT',
whitelisted=True,
body_type='other',
reports=True,
domains=[AgencyDomain(domain=domain, active=True)]))
publish_tasks.agency.delay(publish_tasks.compress_agency(agency),
'created')
return agency
