def edit_source(source_id):
State.load()
source = Source.query.get_or_404(source_id)
form = SourceForm()
if form.validate_on_submit():
if form.test.data:
web_url = form.website.data
Dict = {1: 'kijiji'}
prime_source = prime.Source(module=Dict.get(form.module.data),
module_properties={
'url': web_url,
'botname': "prime"
})
try:
total_ads = prime.test_webui_source(prime_source).total_new_ads
except:
message = "Not a valid source"
else:
message = f"Found {total_ads} new ads" \
if total_ads != 1 else "Found 1 new ad"
finally:
if web_url == "":
message = "Not a valid source"
flash(message, "notification")
else:
source.module = form.module.data
source.name = form.name.data
source.website = form.website.data
source.location = form.location.data
source.range = form.range.data
# source.subreddit = form.subreddit.data
db.session.commit()
State.refresh_sources()
flash('Your source has been updated!', 'top_flash_success')
return redirect(url_for('main.sources', source_id=source.id))
elif request.method == 'GET':
form.module.data = source.module
form.name.data = source.name
form.website.data = source.website
form.location.data = source.location
form.range.data = source.range
# form.subreddit.data = source.subreddit
return render_template('create-source.html',
title='Update Source',
form=form,
legend='Update Source')
def create_source():
State.load()
form = SourceForm()
if form.validate_on_submit():
if form.test.data:
web_url = form.website.data
Dict = {1: 'kijiji'}
prime_source = prime.Source(module=Dict.get(form.module.data),
module_properties={
'url': web_url,
'botname': "prime"
})
try:
total_ads = prime.test_webui_source(prime_source).total_new_ads
except:
message = "Not a valid source"
else:
message = f"Found {total_ads} new ads" \
if total_ads != 1 else "Found 1 new ad"
finally:
if web_url == "":
message = "Not a valid source"
flash(message, "notification")
else:
source = Source(
module=form.module.data,
name=form.name.data,
website=form.website.data,
location=form.location.data,
range=form.range.data,
# subreddit=form.subreddit.data
)
db.session.add(source)
db.session.commit()
State.refresh_sources()
flash('Your source has been saved!', 'top_flash_success')
return redirect(url_for('main.sources'))
return render_template('create-source.html',
title='Create Source',
form=form,
legend='Create Source')