upload.cgi

#!/usr/bin/python
# -*- coding: utf-8 -*-

import cgi
import cgitb; cgitb.enable()
import os, sys
import sssp_name
import subprocess
import boto3
import StringIO

print "Content-Type: text/html"
print
TMP_DIR = "../tmp"

def save_uploaded_file (upload_dir):
  form = cgi.FieldStorage()
  fileitem = form["file"]

  fout = file (os.path.join(upload_dir, os.path.basename(fileitem.filename)), 'wb')
  while 1:
    chunk = fileitem.file.read(100000)
    if not chunk: break
    fout.write (chunk)
  fout.close()
  return(fileitem.filename)

# File Save (TMP_DIR)
file_name=save_uploaded_file(TMP_DIR)
file_pname=os.path.join(TMP_DIR, os.path.basename(file_name))
file_names=[file_pname]

# SAVDI
resp = sssp_name.virus_check(file_names)

if resp >= 2:
  print "Virus found"
elif resp == 1:
  print "Unable to check file"
else:
  s3 = boto3.resource('s3')
  s3.meta.client.upload_file(file_pname, 's3-bucketname', file_name)
  os.remove(file_pname)
  print "Upload ok"