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())
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)