def edit_sample(sample_id):
sample = Sample.query.filter(Sample.id == sample_id).first()
if sample is not None:
versions = CCExtractorVersion.query.all()
# Process or render form
form = EditSampleForm(request.form)
form.version.choices = [(v.id, v.version) for v in versions]
if form.validate_on_submit():
# Store values
upload = sample.upload
upload.notes = form.notes.data
upload.version_id = form.version.data
upload.platform = Platform.from_string(form.platform.data)
upload.parameters = form.parameters.data
g.db.commit()
return redirect(url_for('.sample_by_id', sample_id=sample.id))
if not form.is_submitted():
# Populate form with current set sample values
form.version.data = sample.upload.version.id
form.platform.data = sample.upload.platform.name
form.notes.data = sample.upload.notes
form.parameters.data = sample.upload.parameters
return {
'sample': sample,
'form': form
}
raise SampleNotFoundException('Sample with id %s not found' % sample_id)
def edit_sample(sample_id):
sample = Sample.query.filter(Sample.id == sample_id).first()
if sample is not None:
versions = CCExtractorVersion.query.all()
# Process or render form
form = EditSampleForm(request.form)
form.version.choices = [(v.id, v.version) for v in versions]
if form.validate_on_submit():
# Store values
upload = sample.upload
upload.notes = form.notes.data
upload.version_id = form.version.data
upload.platform = Platform.from_string(form.platform.data)
upload.parameters = form.parameters.data
g.db.commit()
return redirect(url_for('.sample_by_id', sample_id=sample.id))
if not form.is_submitted():
# Populate form with current set sample values
form.version.data = sample.upload.version.id
form.platform.data = sample.upload.platform.name
form.notes.data = sample.upload.notes
form.parameters.data = sample.upload.parameters
return {
'sample': sample,
'form': form
}
raise SampleNotFoundException('Sample with id {id} not found'.format(id=sample_id))
def edit_sample(sample_id):
"""
Edit sample, required admin role.
:param sample_id: id of the sample
:type sample_id: int
:raises SampleNotFoundException: when sample id is not found
:return: form to edit sample
:rtype: dict
"""
sample = Sample.query.filter(Sample.id == sample_id).first()
if sample is not None:
versions = CCExtractorVersion.query.all()
# Process or render form
form = EditSampleForm(request.form)
form.version.choices = [(v.id, v.version) for v in versions]
if form.validate_on_submit():
# Store values
upload = sample.upload
upload.notes = form.notes.data
upload.version_id = form.version.data
upload.platform = Platform.from_string(form.platform.data)
upload.parameters = form.parameters.data
g.db.commit()
g.log.info(f'sample with id: {sample_id} updated')
return redirect(url_for('.sample_by_id', sample_id=sample.id))
if not form.is_submitted():
# Populate form with current set sample values
form.version.data = sample.upload.version.id
form.platform.data = sample.upload.platform.name
form.notes.data = sample.upload.notes
form.parameters.data = sample.upload.parameters
return {'sample': sample, 'form': form}
raise SampleNotFoundException(
'Sample with id {id} not found'.format(id=sample_id))
def process_id(upload_id):
"""
Process the sample that is uploaded to the platform.
:param upload_id: The identity of uploaded file that will be processed
:type upload_id: str
:return: Process progress in form and queue the sample
:rtype: str
"""
from run import config, log
queued_sample = QueuedSample.query.filter(
QueuedSample.id == upload_id).first()
if queued_sample is not None:
if queued_sample.user_id == g.user.id:
versions = CCExtractorVersion.query.all()
form = FinishQueuedSampleForm(request.form)
form.version.choices = [(v.id, v.version) for v in versions]
if form.validate_on_submit():
db_committed = False
repo_folder = config.get('SAMPLE_REPOSITORY', '')
temp_path = os.path.join(repo_folder, 'QueuedFiles',
queued_sample.filename)
final_path = os.path.join(repo_folder, 'TestFiles',
queued_sample.filename)
try:
extension = queued_sample.extension[1:] if len(
queued_sample.extension) > 0 else ""
sample = Sample(queued_sample.sha, extension,
queued_sample.original_name)
g.db.add(sample)
g.db.flush([sample])
uploaded = Upload(g.user.id, sample.id, form.version.data,
Platform.from_string(form.platform.data),
form.parameters.data, form.notes.data)
g.db.add(uploaded)
g.db.delete(queued_sample)
g.db.commit()
db_committed = True
except Exception:
traceback.print_exc()
g.db.rollback()
if db_committed:
if form.report.data == 'y':
data = ""
try:
kvm_name = config.get('KVM_LINUX_NAME', '')
repo = Repo(
os.path.join(repo_folder, 'vm_data', kvm_name,
'unsafe-ccextractor'))
data = repo.git.show(
f"{repo.heads.master}:.github/ISSUE_TEMPLATE.md"
)
except InvalidGitRepositoryError:
log.critical(
"Could not open CCExtractor's repository")
version = CCExtractorVersion.query.filter(
CCExtractorVersion.id ==
form.version.data).first()
data = data.replace("**X.X**", version.version)
data = data.replace("[ ] I have read",
"[X] I have read")
data = data.replace("[ ] I have checked",
"[X] I have checked")
data = data.replace("[ ] I have used",
"[X] I have used")
data = data.replace(
"[ ] I am an active contributor to CCExtractor.",
"[X] I used the platform to submit this issue!")
data = data.replace("`-autoprogram`",
f"`{form.parameters.data}`")
platform = form.platform.data.title()
data = data.replace('[ ] ' + platform,
'[X] ' + platform)
# Remove everything starting from the video links
data = data[:data.find('**Video links**')]
# Append our own content here
sample_link = url_for('sample.sample_by_id',
sample_id=sample.id,
_external=True)
data += f"""
**Sample**
[Sample {sample.id}]({sample_link}) uploaded on the Sample Platform.
**Extra information**
*Notes:*
{form.notes.data}
*Description:*
{form.IssueBody.data}"""
issue_title = f"[BUG] {form.IssueTitle.data}"
issue_data = make_github_issue(
issue_title, data, ['bug', f'sample_{sample.id}'])
if issue_data != 'ERROR':
issue_id = issue_data['number']
issue_title = issue_data['title']
issue_user = issue_data['user']['login']
issue_date = issue_data['created_at']
issue_status = issue_data['state']
issue = Issue(sample.id, issue_id, issue_date,
issue_title, issue_user,
issue_status)
g.db.add(issue)
g.db.commit()
else:
flash(
"Could not submit an issue on GitHub (did you revoke permissions for the platform?)."
" Please submit it manually.")
os.rename(temp_path, final_path)
return redirect(
url_for('sample.sample_by_id', sample_id=sample.id))
return {'form': form, 'queued_sample': queued_sample}
# Raise error
raise QueuedSampleNotFoundException()