def set():
form = SetForm()
if form.validate_on_submit():
if form.key.data and form.value.data:
new_stuff = KeyValueTable(key=form.key.data,
value=form.value.data,
user_id=current_user.id)
db.session.add(new_stuff)
db.session.commit()
flash('Successfully added key and value')
return render_template('set.html', form=form)
def get_pressed():
"""
This function handles retrieving a value from the database when the "GET"
button is pressed. Reads in the entered key and re-renders the home page
with an updated message containing the queried data.
returns:
str: The html represented inside the appropriate html file.
"""
key = dict(request.form.items())['getter_key']
set_form = SetForm(request.form)
get_form = GetForm(request.form)
message = None
try:
# There has to be a better way to do this...
record_index = User.query.filter_by(
email=session['email']).first().records.index(
Record.query.get(key))
value = User.query.filter_by(
email=session['email']).first().records[record_index].value
message = 'The value for "{}" is "{}".'.format(key, value)
except ValueError:
message = 'The key {} does not exist.'.format(key)
except:
message = 'The database is missing or has been renamed.'
if not message:
message = 'A record with that key does not exist.'
return render_template('index.html',
get_form=get_form,
set_form=set_form,
message=message)
def setter_save():
"""
This function handles saving user input to the database when the "SET"
button is pressed.
returns:
str: The html represented inside the appropriate html file.
"""
setter_input = dict(request.form.items())
key = setter_input['setter_key']
value = setter_input['setter_value']
if not Record.query.get(key):
db.session.add(
Record(key, value,
User.query.filter_by(email=session['email']).first()))
db.session.commit()
else:
db.session.delete(Record.query.get(key))
db.session.add(
Record(key, value,
User.query.filter_by(email=session['email']).first()))
db.session.commit()
set_form = SetForm(request.form)
get_form = GetForm(request.form)
message = 'Database was successfully updated!'
return render_template('index.html',
get_form=get_form,
set_form=set_form,
message=message)
def index():
"""
This function builds the view for the main page of the application.
returns:
str: The html represented inside the main html file.
"""
if 'email' not in session:
return redirect(url_for('login'))
set_form = SetForm(request.form)
get_form = GetForm(request.form)
message = 'Use the forms above to interact with the SQL database.'
return render_template('index.html',
get_form=get_form,
set_form=set_form,
essage=message)
def paste(request):
commit_kwargs = {}
if request.user.is_authenticated():
commit_kwargs = {
'anonymous': request.user.preference.default_anonymous
}
if request.method != 'POST':
return render_to_response('paste.html', {
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm(initial=commit_kwargs),
'set_meta_form': SetMetaForm(),
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST, initial=commit_kwargs)
set_meta_form = SetMetaForm(request.POST)
if not (paste_forms.is_valid() and
set_form.is_valid() and
commit_meta_form.is_valid() and
set_meta_form.is_valid()):
return render_to_response('paste.html', {
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form,
'set_meta_form': set_meta_form,
}, RequestContext(request))
owner = get_owner(request, commit_meta_form.cleaned_data, request.user)
description = set_form.cleaned_data.get('description')
private = set_meta_form.cleaned_data.get('private')
allow_edits = set_meta_form.cleaned_data.get('anyone_can_edit')
repo_dir = dirname_from_description(description)
if os.path.isdir(repo_dir):
repo_dir = get_first_nonexistent_filename(repo_dir + '-%d')
os.mkdir(repo_dir)
# Calculate expiration time of set if necessary
exp_option = set_meta_form.cleaned_data.get('expires')
exp_map = {
'day' : timedelta(days=1),
'hour' : timedelta(hours=1),
'month' : timedelta(365/12),
}
exp_time = datetime.utcnow() + exp_map[exp_option] if exp_option in exp_map else None
# Generate a random hash for private access (20-30 characters from letters & numbers)
private_key = ''.join(random.sample(string.ascii_letters + string.digits,
random.randrange(20,30)))
# Create a new paste set so we can reference our paste.
paste_set = Set.objects.create(
views=0,
repo=repo_dir,
owner=owner,
description=description,
private=private,
anyone_can_edit=allow_edits,
private_key=private_key,
expires=exp_time,
)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(
views=0,
parent_set=paste_set,
commit='',
owner=owner
)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
with open(priority_filename, 'w') as priority_file:
for form_index, form in enumerate(paste_forms):
priority_file.write('%s: %s\n' % process_pasted_file(form_index,
form,
repo_dir,
index, commit))
index.add([priority_filename])
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
if paste_set.private:
return redirect('paste_view', pk=paste_set.pk, private_key=paste_set.private_key)
else:
return redirect('paste_view', pk=paste_set.pk)
def live_paste(request):
commit_kwargs = {}
if request.user.is_authenticated():
commit_kwargs = {
'anonymous': request.user.preference.default_anonymous
}
if request.method != 'POST':
return render_to_response('live.html', {
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm(initial=commit_kwargs),
'set_meta_form': SetMetaForm(),
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST, initial=commit_kwargs)
set_meta_form = SetMetaForm(request.POST)
if (not paste_forms.is_valid() or
not set_form.is_valid() or
not commit_meta_form.is_valid() or
not set_meta_form.is_valid()):
return render_to_response('live.html', {
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form,
'set_meta_form': set_meta_form,
}, RequestContext(request))
# Repositories are just a random sequence of letters and digits
# We store the reference repository for editing the pastes.
repo_dir = os.sep.join([
settings.REPO_DIR,
"".join(random.sample(string.letters + string.digits, 15))
])
anonymous = commit_meta_form.cleaned_data.get('anonymous')
os.mkdir(repo_dir)
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
description = set_form.cleaned_data.get('description')
private = set_meta_form.cleaned_data.get('private')
allow_edits = set_meta_form.cleaned_data.get('anyone_can_edit')
# Calculate expiration time of set if necessary
exp_option = set_meta_form.cleaned_data.get('expires')
exp_map = {
'day' : timedelta(days=1),
'hour' : timedelta(hours=1),
'month' : timedelta(365/12),
}
exp_time = datetime.utcnow() + exp_map[exp_option] if exp_option in exp_map else None
# Generate a random hash for private access (20-30 characters from letters & numbers)
private_key = ''.join(random.choice(string.ascii_letters + string.digits) for x in range(random.randrange(20,30)))
# Create a new paste set so we can reference our paste.
paste_set = Set.objects.create(
views=0,
repo=repo_dir,
owner=owner,
description=description,
private=private,
anyone_can_edit=allow_edits,
private_key=private_key,
expires=exp_time,
)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(
views=0,
parent_set=paste_set,
commit='',
owner=owner
)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
priority_file = open(priority_filename, 'w')
for form_index, form in enumerate(paste_forms):
data = form.cleaned_data
filename = data['filename']
language_lex, language = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
if not len(filename):
filename = 'paste'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([
repo_dir,
filename
])
filename_absolute += ext
filename_abs_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
filename += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_abs_base, i, ext)
filename = '%s-%d%s' % (filename_base, i, ext)
i += 1
cleaned = []
paste = paste.encode('UTF-8')
for line in paste.split('\n'):
line = line.rstrip()
cleaned.append(line)
paste = '\n'.join(cleaned)
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
priority_file.write('%s: %s\n' % (filename, data['priority']))
paste = smart_unicode(paste)
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
lex = globals()[language_lex]
paste_formatted = highlight(
paste,
lex(),
HtmlFormatter(
style='friendly',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True)
)
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(
filename=filename,
absolute_path=filename_absolute,
paste=paste,
priority=data['priority'],
paste_formatted=paste_formatted,
language=data['language'],
revision=commit
)
# Add a priority file
priority_file.close()
index.add([priority_filename])
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
if not paste_set.private:
return redirect('paste_view', pk=paste_set.pk)
else:
return redirect('paste_view', pk=paste_set.pk, private_key=paste_set.private_key)
return render_to_response('live.html')
def paste(request):
commit_kwargs = {}
user = request.user
if user.is_authenticated():
if settings.ALLOW_ANONYMOUS_POSTS:
commit_kwargs = {'anonymous': user.preference.default_anonymous}
elif not settings.ALLOW_ANONYMOUS_POSTS:
return redirect('login')
elif not settings.ALLOW_ANONYMOUS_ACCESS:
return redirect('login')
if request.method != 'POST':
return render_to_response(
'paste.html', {
'allow_anonymous_posts': settings.ALLOW_ANONYMOUS_POSTS,
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm(initial=commit_kwargs),
'set_meta_form': SetMetaForm(),
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST, initial=commit_kwargs)
set_meta_form = SetMetaForm(request.POST)
if not (paste_forms.is_valid() and set_form.is_valid()
and commit_meta_form.is_valid() and set_meta_form.is_valid()):
return render_to_response(
'paste.html', {
'allow_anonymous_posts': settings.ALLOW_ANONYMOUS_POSTS,
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form,
'set_meta_form': set_meta_form,
}, RequestContext(request))
owner = get_owner(request, commit_meta_form.cleaned_data, request.user)
description = set_form.cleaned_data.get('description')
private = set_meta_form.cleaned_data.get('private')
allow_edits = set_meta_form.cleaned_data.get('anyone_can_edit')
repo_dir = dirname_from_description(description)
if not len(description):
repo_dir = dirname_from_description(''.join(
random.sample(string.ascii_letters + string.digits,
random.randrange(20, 30))))
if os.path.isdir(repo_dir):
repo_dir = get_first_nonexistent_filename(repo_dir + '-%d')
os.mkdir(repo_dir)
# Calculate expiration time of set if necessary
exp_option = set_meta_form.cleaned_data.get('expires')
exp_map = {
'day': timedelta(days=1),
'hour': timedelta(hours=1),
'month': timedelta(365 / 12),
}
exp_time = datetime.utcnow(
) + exp_map[exp_option] if exp_option in exp_map else None
# Generate a random hash for private access (20-30 characters from letters & numbers)
private_key = ''.join(
random.sample(string.ascii_letters + string.digits,
random.randrange(20, 30)))
# Create a new paste set so we can reference our paste.
paste_set = Set.objects.create(
views=0,
repo=repo_dir,
owner=owner,
description=description,
private=private,
anyone_can_edit=allow_edits,
private_key=private_key,
expires=exp_time,
)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(views=0,
parent_set=paste_set,
commit='',
owner=owner)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
with codecs.open(priority_filename, 'w', "utf-8-sig") as priority_file:
for form_index, form in enumerate(paste_forms):
priority_file.write(
'%s: %s\n' %
process_pasted_file(form_index, form, repo_dir, index, commit))
index.add([priority_filename])
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
if paste_set.private:
return redirect('paste_view',
pk=paste_set.pk,
private_key=paste_set.private_key)
else:
return redirect('paste_view', pk=paste_set.pk)
def live_paste(request):
commit_kwargs = {}
if request.user.is_authenticated():
commit_kwargs = {
'anonymous': request.user.preference.default_anonymous
}
if request.method != 'POST':
return render_to_response(
'live.html', {
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm(initial=commit_kwargs),
'set_meta_form': SetMetaForm(),
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST, initial=commit_kwargs)
set_meta_form = SetMetaForm(request.POST)
if (not paste_forms.is_valid() or not set_form.is_valid()
or not commit_meta_form.is_valid()
or not set_meta_form.is_valid()):
return render_to_response(
'live.html', {
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form,
'set_meta_form': set_meta_form,
}, RequestContext(request))
# Repositories are just a random sequence of letters and digits
# We store the reference repository for editing the pastes.
repo_dir = os.sep.join([
settings.REPO_DIR,
"".join(random.sample(string.letters + string.digits, 15))
])
anonymous = commit_meta_form.cleaned_data.get('anonymous')
os.mkdir(repo_dir)
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
description = set_form.cleaned_data.get('description')
private = set_meta_form.cleaned_data.get('private')
allow_edits = set_meta_form.cleaned_data.get('anyone_can_edit')
# Calculate expiration time of set if necessary
exp_option = set_meta_form.cleaned_data.get('expires')
exp_map = {
'day': timedelta(days=1),
'hour': timedelta(hours=1),
'month': timedelta(365 / 12),
}
exp_time = datetime.utcnow(
) + exp_map[exp_option] if exp_option in exp_map else None
# Generate a random hash for private access (20-30 characters from letters & numbers)
private_key = ''.join(
random.choice(string.ascii_letters + string.digits)
for x in range(random.randrange(20, 30)))
# Create a new paste set so we can reference our paste.
paste_set = Set.objects.create(
views=0,
repo=repo_dir,
owner=owner,
description=description,
private=private,
anyone_can_edit=allow_edits,
private_key=private_key,
expires=exp_time,
)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(views=0,
parent_set=paste_set,
commit='',
owner=owner)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
priority_file = open(priority_filename, 'w')
for form_index, form in enumerate(paste_forms):
data = form.cleaned_data
filename = data['filename']
language_lex, language = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
if not len(filename):
filename = 'paste'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([repo_dir, filename])
filename_absolute += ext
filename_abs_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
filename += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_abs_base, i, ext)
filename = '%s-%d%s' % (filename_base, i, ext)
i += 1
cleaned = []
paste = paste.encode('UTF-8')
for line in paste.split('\n'):
line = line.rstrip()
cleaned.append(line)
paste = '\n'.join(cleaned)
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
priority_file.write('%s: %s\n' % (filename, data['priority']))
paste = smart_unicode(paste)
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
lex = globals()[language_lex]
paste_formatted = highlight(
paste, lex(),
HtmlFormatter(style='friendly',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True))
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(filename=filename,
absolute_path=filename_absolute,
paste=paste,
priority=data['priority'],
paste_formatted=paste_formatted,
language=data['language'],
revision=commit)
# Add a priority file
priority_file.close()
index.add([priority_filename])
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
if not paste_set.private:
return redirect('paste_view', pk=paste_set.pk)
else:
return redirect('paste_view',
pk=paste_set.pk,
private_key=paste_set.private_key)
return render_to_response('live.html')
def paste_edit(request, pk, paste_set, private_key=None):
requested_commit = request.GET.get('commit')
# You can technically modify anything in history and update it
if requested_commit is None:
commit = paste_set.commit_set.latest('id')
else:
commit = get_object_or_404(Commit,
parent_set=paste_set,
commit=requested_commit)
previous_files = []
for f in commit.paste_set.all():
previous_files.append(os.path.basename(f.absolute_path))
# Populate our initial data
initial_data = []
for paste in commit.paste_set.all():
initial_data.append({
'filename': paste.filename,
'paste': paste.paste,
'language': paste.language,
})
initial_set_meta = {
'private': paste_set.private,
'expires': paste_set.expires or "never",
'anyone_can_edit': paste_set.anyone_can_edit
}
#TODO: turn this into a template tag and allow template to do conversion
original_expires_time = paste_set.expires
expires_time = None
if original_expires_time:
if timezone.is_naive(original_expires_time):
original_expires_time = original_expires_time.replace(
tzinfo=timezone.utc)
expires_time = original_expires_time.astimezone(
timezone.get_current_timezone())
if request.method != 'POST':
set_form = None
if request.user == paste_set.owner:
set_form_initial = {'description': paste_set.description}
set_form = SetForm(initial=set_form_initial)
return render_to_response(
'paste.html', {
'forms': PasteSetEdit(initial=initial_data),
'set_form': set_form,
'commit_meta_form': CommitMetaForm(),
'set_meta_form': SetMetaForm(initial=initial_set_meta),
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
set_form = None
set_meta_form = None
forms = PasteSetEdit(request.POST, initial=initial_data)
commit_meta_form = CommitMetaForm(request.POST)
form_list = [forms, commit_meta_form]
if request.user == paste_set.owner:
set_form = SetForm(request.POST)
set_meta_form = SetMetaForm(request.POST)
form_list += [set_form, set_meta_form]
if not all(map(lambda x: x.is_valid(), form_list)):
return render_to_response(
'paste.html', {
'forms': forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form,
'set_meta_form': set_meta_form,
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
# Update the repo
repo_dir = paste_set.repo
repo = git.Repo(repo_dir)
index = repo.index
anonymous = commit_meta_form.cleaned_data['anonymous']
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
if set_form:
fdata = set_form.cleaned_data
paste_set.description = fdata['description']
if set_meta_form:
fdata = set_meta_form.cleaned_data
paste_set.private = fdata.get('private')
paste_set.anyone_can_edit = fdata.get('anyone_can_edit')
paste_set.save()
commit = Commit.objects.create(views=0,
parent_set=paste_set,
commit='',
owner=owner)
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
form_files = []
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
with codecs.open(priority_filename, 'w', "utf-8-sig") as priority_file:
for form_index, form in enumerate(forms):
filename, priority = process_pasted_file(form_index, form,
repo_dir, index, commit,
True)
form_files.append(filename)
priority_file.write('%s: %s\n' % (filename, 'priority'))
# Create the commit from the index
intersected = set(form_files).intersection(previous_files)
removed_files = list(set(previous_files) - intersected)
for f in removed_files:
index.remove([os.sep.join([repo_dir, f])])
index.add([priority_filename])
new_commit = index.commit('Modified.')
commit.commit = new_commit
commit.diff = _git_diff(new_commit, repo)
commit.save()
if not paste_set.private:
return redirect('paste_view', pk=paste_set.pk)
else:
return redirect('paste_view',
pk=paste_set.pk,
private_key=paste_set.private_key)
def paste_edit(request, pk, paste_set, private_key=None):
requested_commit = request.GET.get('commit')
# You can technically modify anything in history and update it
if requested_commit is None:
commit = paste_set.commit_set.latest('id')
else:
commit = get_object_or_404(Commit,
parent_set=paste_set,
commit=requested_commit)
previous_files = []
for f in commit.paste_set.all():
previous_files.append(os.path.basename(f.absolute_path))
# Populate our initial data
initial_data = []
for paste in commit.paste_set.all():
initial_data.append({
'filename': paste.filename,
'paste': paste.paste,
'language': paste.language,
})
initial_set_meta = {
'private': paste_set.private,
'expires': paste_set.expires,
}
#TODO: turn this into a template tag and allow template to do conversion
original_expires_time = paste_set.expires
expires_time = None
if original_expires_time:
if timezone.is_naive(original_expires_time):
original_expires_time = original_expires_time.replace(
tzinfo=timezone.utc)
expires_time = original_expires_time.astimezone(
timezone.get_current_timezone())
if request.method != 'POST':
set_form = None
if request.user == paste_set.owner:
set_form_initial = {'description': paste_set.description}
set_form = SetForm(initial=set_form_initial)
return render_to_response(
'paste.html', {
'forms': PasteSetEdit(initial=initial_data),
'set_form': set_form,
'commit_meta_form': CommitMetaForm(),
'set_meta_form': SetMetaForm(initial=initial_set_meta),
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
forms = PasteSetEdit(request.POST, initial=initial_data)
commit_meta_form = CommitMetaForm(request.POST)
set_form = None
set_meta_form = None
if request.user == paste_set.owner:
set_form = SetForm(request.POST)
set_meta_initial = {
'expires': 'never'
} # to stop validation error, not used. is there a better way to do this?
set_meta_form = SetMetaForm(request.POST, initial=set_meta_initial)
if not forms.is_valid() or not commit_meta_form.is_valid() or (
set_form is not None
and not set_form.is_valid()) or (set_meta_form is not None
and not set_meta_form.is_valid()):
return render_to_response(
'paste.html', {
'forms': forms,
'set_form': set_form,
'commit_meta_form': CommitMetaForm(),
'set_meta_form': SetMetaForm(initial=initial_set_meta),
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
# Update the repo
repo_dir = paste_set.repo
repo = git.Repo(repo_dir)
index = repo.index
anonymous = commit_meta_form.cleaned_data['anonymous']
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
if set_form:
paste_set.description = set_form.cleaned_data['description']
paste_set.private = set_meta_form.cleaned_data.get('private')
paste_set.save()
commit = Commit.objects.create(views=0,
parent_set=paste_set,
commit='',
owner=owner)
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
form_files = []
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
priority_file = open(priority_filename, 'w')
for form_index, form in enumerate(forms):
data = form.cleaned_data
filename = data['filename']
language_lex, language = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
no_filename = False
if not len(filename):
no_filename = True
filename = 'a-lonely-file'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([repo_dir, filename])
filename_absolute += ext
filename_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
if no_filename:
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_base, i, ext)
i += 1
form_files.append(os.path.basename(filename_absolute))
cleaned = []
paste = paste.encode('UTF-8')
for line in paste.split('\n'):
line = line.rstrip()
cleaned.append(line)
paste = '\n'.join(cleaned)
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
priority_file.write('%s: %s\n' % (filename, data['priority']))
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
paste = smart_unicode(paste)
lex = globals()[language_lex]
paste_formatted = highlight(
paste, lex(),
HtmlFormatter(style='colorful',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True))
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(filename=filename,
absolute_path=filename_absolute,
paste=paste,
priority=data['priority'],
paste_formatted=paste_formatted,
language=data['language'],
revision=commit)
# Create the commit from the index
intersected = set(form_files).intersection(previous_files)
removed_files = list(set(previous_files) - intersected)
for f in removed_files:
index.remove([os.sep.join([repo_dir, f])])
priority_file.close()
index.add([priority_filename])
new_commit = index.commit('Modified.')
commit.commit = new_commit
commit.diff = _git_diff(new_commit, repo)
commit.save()
if not paste_set.private or user_owns_paste(paste_set, request.user):
return redirect('paste_view', pk=paste_set.pk)
else:
return redirect('paste_view',
pk=paste_set.pk,
private_key=paste_set.private_key)
def paste_edit(request, pk, paste_set, private_key=None):
requested_commit = request.GET.get('commit')
# You can technically modify anything in history and update it
if requested_commit is None:
commit = paste_set.commit_set.latest('id')
else:
commit = get_object_or_404(Commit,
parent_set=paste_set, commit=requested_commit)
previous_files = []
for f in commit.paste_set.all():
previous_files.append(os.path.basename(f.absolute_path))
# Populate our initial data
initial_data = []
for paste in commit.paste_set.all():
initial_data.append({
'filename': paste.filename,
'paste': paste.paste,
'language': paste.language,
})
initial_set_meta = {
'private': paste_set.private,
'expires': paste_set.expires,
}
#TODO: turn this into a template tag and allow template to do conversion
original_expires_time = paste_set.expires
expires_time = None
if original_expires_time:
if timezone.is_naive(original_expires_time):
original_expires_time = original_expires_time.replace(tzinfo=timezone.utc)
expires_time = original_expires_time.astimezone(timezone.get_current_timezone())
if request.method != 'POST':
set_form = None
if request.user == paste_set.owner:
set_form_initial = {'description': paste_set.description}
set_form = SetForm(initial=set_form_initial)
return render_to_response('paste.html', {
'forms': PasteSetEdit(initial=initial_data),
'set_form': set_form,
'commit_meta_form': CommitMetaForm(),
'set_meta_form': SetMetaForm(initial=initial_set_meta),
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
forms = PasteSetEdit(request.POST, initial=initial_data)
commit_meta_form = CommitMetaForm(request.POST)
set_form = None
set_meta_form = None
if request.user == paste_set.owner:
set_form = SetForm(request.POST)
set_meta_initial = {'expires': 'never'} # to stop validation error, not used. is there a better way to do this?
set_meta_form = SetMetaForm(request.POST, initial=set_meta_initial)
if not forms.is_valid() or not commit_meta_form.is_valid() or (
set_form is not None and not set_form.is_valid() ) or (
set_meta_form is not None and not set_meta_form.is_valid()):
return render_to_response('paste.html', {
'forms': forms,
'set_form': set_form,
'commit_meta_form': CommitMetaForm(),
'set_meta_form': SetMetaForm(initial=initial_set_meta),
'expires_time': expires_time,
'editing': True,
}, RequestContext(request))
# Update the repo
repo_dir = paste_set.repo
repo = git.Repo(repo_dir)
index = repo.index
anonymous = commit_meta_form.cleaned_data['anonymous']
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
if set_form:
paste_set.description = set_form.cleaned_data['description']
paste_set.private = set_meta_form.cleaned_data.get('private')
paste_set.save()
commit = Commit.objects.create(
views=0,
parent_set=paste_set,
commit='',
owner=owner
)
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
form_files = []
priority_filename = os.sep.join([repo_dir, 'priority.txt'])
priority_file = open(priority_filename, 'w')
for form_index, form in enumerate(forms):
data = form.cleaned_data
filename = data['filename']
language_lex, language = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
no_filename = False
if not len(filename):
no_filename = True
filename = 'a-lonely-file'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([
repo_dir,
filename
])
filename_absolute += ext
filename_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
if no_filename:
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_base, i, ext)
i += 1
form_files.append(os.path.basename(filename_absolute))
cleaned = []
paste = paste.encode('UTF-8')
for line in paste.split('\n'):
line = line.rstrip()
cleaned.append(line)
paste = '\n'.join(cleaned)
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
priority_file.write('%s: %s\n' % (filename, data['priority']))
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
paste = smart_unicode(paste)
lex = globals()[language_lex]
paste_formatted = highlight(
paste,
lex(),
HtmlFormatter(
style='colorful',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True)
)
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(
filename=filename,
absolute_path=filename_absolute,
paste=paste,
priority=data['priority'],
paste_formatted=paste_formatted,
language=data['language'],
revision=commit
)
# Create the commit from the index
intersected = set(form_files).intersection(previous_files)
removed_files = list(set(previous_files) - intersected)
for f in removed_files:
index.remove([os.sep.join([
repo_dir,
f
])])
priority_file.close()
index.add([priority_filename])
new_commit = index.commit('Modified.')
commit.commit = new_commit
commit.diff = _git_diff(new_commit, repo)
commit.save()
if not paste_set.private or user_owns_paste(paste_set, request.user):
return redirect('paste_view', pk=paste_set.pk)
else:
return redirect('paste_view', pk=paste_set.pk, private_key=paste_set.private_key)
def paste(request):
if request.method != 'POST':
return render_to_response(
'paste.html', {
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm()
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST)
if (not paste_forms.is_valid() or not set_form.is_valid()
or not commit_meta_form.is_valid()):
return render_to_response(
'paste.html', {
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form
}, RequestContext(request))
# Repositories are just a random sequence of letters and digits
# We store the reference repository for editing the pastes.
repo_dir = os.sep.join([
settings.REPO_DIR,
"".join(random.sample(string.letters + string.digits, 15))
])
anonymous = commit_meta_form.cleaned_data['anonymous']
os.mkdir(repo_dir)
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
# Create a new paste set so we can reference our paste.
description = set_form.cleaned_data.get('description')
paste_set = Set.objects.create(repo=repo_dir,
owner=owner,
description=description)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(parent_set=paste_set,
commit='',
owner=owner)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
for form_index, form in enumerate(paste_forms):
data = form.cleaned_data
filename = data['filename']
language_lex, language = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
if not len(filename):
filename = 'a-lonely-file'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([repo_dir, filename])
filename_absolute += ext
filename_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_base, i, ext)
i += 1
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
lex = globals()[language_lex]
paste_formatted = highlight(
paste, lex(),
HtmlFormatter(style='colorful',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True))
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(filename=filename,
absolute_path=filename_absolute,
paste=paste,
paste_formatted=paste_formatted,
language=data['language'],
revision=commit)
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
return redirect('paste_view', pk=paste_set.pk)
def paste(request):
if request.method != 'POST':
return render_to_response('paste.html', {
'forms': PasteSet(),
'set_form': SetForm(),
'commit_meta_form': CommitMetaForm()
}, RequestContext(request))
paste_forms = PasteSet(request.POST)
set_form = SetForm(request.POST)
commit_meta_form = CommitMetaForm(request.POST)
if (not paste_forms.is_valid() or
not set_form.is_valid() or
not commit_meta_form.is_valid()):
return render_to_response('paste.html', {
'forms': paste_forms,
'set_form': set_form,
'commit_meta_form': commit_meta_form
}, RequestContext(request))
# Repositories are just a random sequence of letters and digits
# We store the reference repository for editing the pastes.
repo_dir = os.sep.join([
settings.REPO_DIR,
"".join(random.sample(string.letters + string.digits, 15))
])
anonymous = commit_meta_form.cleaned_data['anonymous']
os.mkdir(repo_dir)
owner = None
if request.user.is_authenticated() and not anonymous:
owner = request.user
# Create a new paste set so we can reference our paste.
description = set_form.cleaned_data.get('description')
paste_set = Set.objects.create(
repo=repo_dir,
owner=owner,
description=description
)
# Yes, this is horrible. I know. But there is a bug with Python Git.
# See: https://github.com/gitpython-developers/GitPython/issues/39
os.environ['USER'] = "******"
if owner:
os.environ['USER'] = owner.username
# Initialize a commit, git repository, and pull the current index.
commit = Commit.objects.create(
parent_set=paste_set,
commit='',
owner=owner
)
git_repo = git.Repo.init(repo_dir)
index = git_repo.index
# We enumerate over the forms so we can have a way to reference
# the line numbers in a unique way relevant to the pastes.
for form_index, form in enumerate(paste_forms):
data = form.cleaned_data
filename = data['filename']
language, language_lex = data['language'].split(';')
paste = data['paste']
# If we don't specify a filename, then obviously it is lonely
if not len(filename):
filename = 'a-lonely-file'
# Construct a more logical filename for our commit
filename_base, ext = os.path.splitext(filename)
filename_slugify = slugify(filename[:len(ext)])
filename_absolute = os.sep.join([
repo_dir,
filename
])
filename_absolute += ext
filename_base, ext = os.path.splitext(filename_absolute)
# If no extension was specified in the file, then we can append
# the extension from the lexer.
if not len(ext):
filename_absolute += language
ext = language
# Gists doesn't allow for the same filename, we do.
# Just append a number to the filename and call it good.
i = 1
while os.path.exists(filename_absolute):
filename_absolute = '%s-%d%s' % (filename_base, i, ext)
i += 1
# Open the file, write the paste, call it good.
f = open(filename_absolute, "w")
f.write(paste)
f.close()
# This is a bit nasty and a get_by_ext something exist in pygments.
# However, globals() is just much more fun.
lex = globals()[language_lex]
paste_formatted = highlight(
paste,
lex(),
HtmlFormatter(
style='colorful',
linenos='table',
lineanchors='line-%s' % form_index,
anchorlinenos=True)
)
# Add the file to the index and create the paste
index.add([filename_absolute])
p = Paste.objects.create(
filename=filename,
absolute_path=filename_absolute,
paste=paste,
paste_formatted=paste_formatted,
language=data['language'],
revision=commit
)
# Create the commit from the index
new_commit = index.commit('Initial paste.')
commit.commit = new_commit
commit.save()
return redirect('paste_view', pk=paste_set.pk)