def setUp(self):
models.initialize()
Entry.create(timestamp=datetime.datetime(2018, 7, 2, 8),
user_name='White Diamond',
task_name='Unknown',
task_minutes=1200,
task_notes="mystery")
def test_edit_entry_bad_then_good(self):
with unittest.mock.patch('builtins.input',
side_effect=['Spam',
'fourtwo',
'',
'42',
'Bey Day',
'',
'11/11/2011',
'n',
'y',
'']):
Entry.create(
timestamp=datetime.datetime(2018, 1, 1, 1),
user_name='Yellow Diamond',
task_name='Stomp RQ',
task_minutes=1,
task_notes="mystery"
)
e_dict = {
'user_name': 'Yellow Diamond',
'task_name': 'Stomp RQ',
'task_minutes': 1,
'task_notes': "mystery",
'timestamp': datetime.datetime(2018, 1, 1, 1)
}
self.assertIsNone(wlui.edit_entry(e_dict))
CardCatalog().delete_entry(
{'user_name': 'Yellow Diamond',
'task_name': 'Spam',
'task_minutes': 42,
'task_notes': "mystery",
'timestamp': datetime.datetime(2011, 11, 11, 1)}
)
def new():
# Create the form
form = EntryForm(request.form)
# If we're posting and the form is filled out correctly
# Create the new entry
# Otherwise return the empty form
if request.method == 'POST' and form.validate():
Entry.create(**form.data)
return redirect('/')
return render_template('new.html', form=form)
def add():
'''Add an entry using a form and store it'''
form = forms.AddEditForm()
if form.validate_on_submit():
Entry.create(title=form.title.data.strip(),
date=form.date.data,
time_spent=form.time_spent.data,
learned=form.learned.data.strip(),
resources=form.resources.data.strip(),
slug=slugify(form.title.data.strip()),
tags=form.tags.data.strip())
flash("Entry added", "success")
return redirect(url_for('index'))
return render_template('new.html', form=form)
def test_entry_list(self):
entry_data = {
'title': 'testing',
'content': 'testing entry',
'resources': 'resources',
'time_spent': 60
}
UserModelTestCase.create_users(1)
entry_data['user'] = User.select().get()
Entry.create(**entry_data)
rv = self.app.get('/')
self.assertNotIn('no entries yet', rv.get_data(as_text=True))
self.assertIn(entry_data['title'], rv.get_data(as_text=True))
def test_details_menu(self):
with test_database(TEST_DB, (User, Entry)):
UserModelTestCase.create_users()
user = User.select().get()
Entry.create(user=user,
title='Coding',
entry_date='November the twentieth',
time_spent='Fifteen minutes',
learned='Mocking a database with test_database()',
resources='various online things.')
entry = Entry.select().get()
rv = self.app.get('/details/{}'.format(entry.id))
self.assertIn("edit entry", rv.get_data(as_text=True).lower())
self.assertIn("fifteen", rv.get_data(as_text=True).lower())
self.assertIn("coding", rv.get_data(as_text=True).lower())
def test_entry_creation(self):
"""This tests if an Entry can be created."""
with test_database(TEST_DB, (User, Entry)):
UserModelTestCase.create_users()
user = User.select().get()
Entry.create(user=user,
title='Coding',
entry_date='November the twentieth',
time_spent='Fifteen minutes',
learned='Mocking a database with test_database()',
resources='various online things.')
entry = Entry.select().get()
self.assertEqual(Entry.select().count(), 1)
self.assertEqual(entry.title, 'Coding')
self.assertEqual(entry.user, user)
def save_new(self, user_name, task_name, task_minutes, task_notes):
"""Save a new entry."""
# Under test.
try:
Entry.create(user_name=user_name,
task_name=task_name,
task_minutes=task_minutes,
task_notes=task_notes)
except ValueError:
print('Unexpected values found in entry.')
print('Please carefully follow all instructions.')
print('Unable to save.')
return False
else:
print('Entry should be added!')
return True
def test_entry_creation(self):
UserModelTestCase.create_users()
user = User.select().get()
Entry.create(
title='testing',
content='testing entry',
resources='resources test',
time_spent=60,
user=user
)
entry = Entry.select().get()
self.assertEqual(
Entry.select().count(),
1
)
self.assertEqual(entry.user, user)
def new_view(parent_id=None):
user = Session.get_user()
if parent_id and Entry.select().where(Entry.id == parent_id, Entry.is_active == True).exists():
parent = Entry.get(Entry.id == parent_id, Entry.is_active == True)
else:
parent = None
title = views.form_get('title', 'no title')
body = views.form_get('body', '')
Entry.create(
user=user,
parent=parent,
title=title,
body=body,
)
return redirect(URLS['index'])
def setUp(self):
models.initialize()
self.view_test = View.__new__(View)
self.deletion_entry = Entry.create(name="Test Name",
task="Test Task",
minutes=120,
notes="Test Notes")
self.entries_test = Entry.get(Entry.select().where(
Entry.task.contains("test task")))
self.entry = self.entries_test
def create():
if request.method == "POST":
if request.form.get("title") and request.form.get("content"):
try:
entry = Entry.create(
title=request.form["title"],
content=request.form["content"],
archived=request.form.get("archived") or False,
)
tags = request.form["tags"].split()
# present is a check to see if the tag exists
present = 0
# add or create tags
for tag in tags:
for entrytag in entry.tags:
if tag == entrytag.tag:
present = 1
if present == 0:
try:
thistag = Tag.get(Tag.tag == tag)
entry.tags.add(thistag)
except:
tag_obj, was_created = Tag.create_or_get(tag=tag)
EntryTags.create(tag=tag_obj, entry=entry)
present = 0
flash("Entry created successfully.", "success")
return redirect(url_for("detail", slug=entry.slug))
except:
flash("Note title already exists", "danger")
return render_template("create.html")
# TODO Refactor the below and above to make it more DRY or not
# to need to display seconds (e.g. add some kind of suffix if entry
# already exists)
elif request.form.get("content"):
entry = Entry.create(
title="{:%a %d %b %Y at %H:%M:%S}".format(datetime.datetime.now()), content=request.form["content"]
)
flash("Note created successfully.", "success")
return redirect(url_for("detail", slug=entry.slug))
else:
flash("Content is required.", "danger")
return render_template("create.html")
def create():
if request.method == 'POST':
if request.form.get('title') and request.form.get('content'):
try:
entry = Entry.create(
title=request.form['title'],
content=request.form['content'],
archived=request.form.get('archived') or False)
tags = request.form['tags'].split()
# present is a check to see if the tag exists
present = 0
# add or create tags
for tag in tags:
for entrytag in entry.tags:
if tag == entrytag.tag:
present = 1
if present == 0:
try:
thistag = Tag.get(Tag.tag == tag)
entry.tags.add(thistag)
except:
tag_obj, was_created = Tag.create_or_get(tag=tag)
EntryTags.create(tag=tag_obj, entry=entry)
present = 0
flash('Entry created successfully.', 'success')
return redirect(url_for('detail', slug=entry.slug))
except:
flash('Note title already exists', 'danger')
return render_template('create.html')
# TODO Refactor the below and above to make it more DRY or not
# to need to display seconds (e.g. add some kind of suffix if entry
# already exists)
elif request.form.get('content'):
entry = Entry.create(
title="{:%a %d %b %Y at %H:%M:%S}".format(datetime.datetime.now()),
content=request.form['content'])
flash('Note created successfully.', 'success')
return redirect(url_for('detail', slug=entry.slug))
else:
flash('Content is required.', 'danger')
return render_template('create.html')
def test_by_staff_fol(self, mock_ve):
with unittest.mock.patch('builtins.input',
side_effect=['n',
'Yellow',
0]):
Entry.create(
timestamp=datetime.datetime(2018, 1, 1, 1),
user_name='Yellow Diamond',
task_name='Stomp RQ',
task_minutes=1,
task_notes="mystery"
)
wlui.by_staff()
mock_ve.assert_called_with(bycat='name', target='Yellow Diamond')
CardCatalog().delete_entry(
{'user_name': 'Yellow Diamond',
'task_name': 'Stomp RQ',
'task_minutes': 1,
'task_notes': "mystery",
'timestamp': datetime.datetime(2018, 1, 1, 1)}
)
def post(self):
page = self.get_int_param("page")
location = self.get_int_param("location")
book_key_name = self.request.get("book")
if page or location:
try:
entry = Entry.create(book = Book.get_by_key_name(book_key_name),
reader = users.get_current_user(),
page = page,
location = location)
Progress.create(entry)
except ValueError, error:
self.set_cookie("flash_message", error)
def post(self, request, *args, **kwargs):
form = TextForm(request.POST)
if form.is_valid():
cd = form.cleaned_data
title = cd.get('title')
text = cd.get('text')
cover = cd.get('cover', '')
entry = Entry.create(title, text, cover)
return HttpResponseRedirect(
reverse('detail', kwargs={'_id': str(entry.id)}))
return render_to_response('create.html', {
'form': form,
}, RequestContext(request))
def create():
if request.method == 'POST':
if request.form.get('title') and request.form.get('content'):
entry = Entry.create(
title=request.form['title'],
content=request.form['content'],
published=request.form.get('published') or False)
flash('Entry created successfully.', 'success')
if entry.published:
return redirect(url_for('detail', slug=entry.slug))
else:
return redirect(url_for('edit', slug=entry.slug))
else:
flash('Title and Content are required.', 'danger')
return render_template('create.html')
def create():
"""Create new blog Entry"""
if request.method == 'POST':
if request.form.get('title') and request.form.get('content'):
entry = Entry.create(title=request.form.get('title'),
content=request.form.get('content'),
published=request.form.get('published')
or False)
flash('Entry created successfully!', 'success')
if entry.published:
return redirect(url_for('detail', slug=entry.slug))
else:
return redirect(url_for('edit', slug=entry.slug))
else:
flash('Title and Content are required!', 'danger')
return render_template('create.html')
def create():
form = EntryForm()
if form.validate_on_submit():
try:
entry = Entry.create(
title=form.title.data,
date=form.date.data,
time=form.time.data,
learned=form.learned.data,
resources=form.resources.data,
slug=safe_slugify(form.title.data),
user=g.user.id
)
process_tags(form.tags.data, entry)
except IntegrityError:
flash("An entry with that title already exists.")
else:
return redirect(url_for('index'))
else:
flash_errors(form)
return render_template('new.html', form=form)
if __name__ == '__main__':
initialize()
if not User.select():
for data in testdata.test_users:
try:
User.create_user(
username=data['username'],
email=data['email'],
password=data['password'],
is_admin=data['is_admin']
)
except (IntegrityError, ValueError):
pass
if not Entry.select():
for data in testdata.test_entries:
entry = Entry.create(
title=data['title'],
time=data['time'],
learned=data['learned'],
resources=data['resources'],
slug=safe_slugify(data['title']),
user=data['user_id']
)
if 'tags' in data:
process_tags(data['tags'], entry)
app.run(debug=DEBUG, host=HOST, port=PORT)
def add_entry(self, name, task, minutes, notes):
Entry.create(name=name, task=task, minutes=minutes, notes=notes)
def create_entry(blog, title, content, pub_date=None):
return Entry.create(blog=blog,
title=title,
content=content,
pub_date=pub_date or datetime.datetime.now())
def create_entry(blog, title, content, pub_date=None):
return Entry.create(blog=blog, title=title, content=content,
pub_date=pub_date or datetime.datetime.now())
import csv
import subprocess
import time
from itertools import islice
from mongoengine import connect
from models import Entry
connect(db='gpu_usage', host='mongodb://gpu4:27017/cin_cluster')
while True:
subprocess.call(['./shell/nvidia-smi.sh'], shell=True)
with open('data/gpu.csv', 'r', encoding='utf-8') as csvfile:
results = csv.reader(csvfile, delimiter=',')
rows = islice(results, 1, None) # skip headers
entries = [Entry.create(row) for row in rows]
for entry in entries:
print(entry)
entry.save()
time.sleep(5)
def new_entry():
form = forms.EntryForm()
if form.validate_on_submit():
Entry.create(**dict(form.data.items()))
return redirect(url_for('index'))
return render_template('new.html', form=form)
