def __init__(self, data, source): self.data = data self.source = source if not Event.query.filter( Event.name == "{} scraping".format(self.source)).first(): # add source obj = Source() obj.name = self.source db.session.add(obj) db.session.commit() # create event self.add_event({ "name": "{} scraping".format(self.source), "details": "Observables mined from {}".format(self.source), "confidence": 10, "source": source, "tlp": "Green", "impact": "Low", "likelihood": "Low" })
def setUp(self): # create source source = Source.find_source_by_id(1) if source is None: source = Source(id=1, name="source1", url="url1") db.session.add(source) db.session.commit() else: source.name = "source1" source.url = "url1" db.session.flush() db.session.commit()
def source_create(): form = SourceCreateForm() if form.validate_on_submit(): source = Source() source.sid = form.sid.data source.name = form.name.data source.url = form.url.data db.session.add(source) db.session.commit() return redirect(url_for('sources')) return render_template('source_create.html', form=form)