# UploadFormat Parser -- test version
#                                       -- ideamonk

class Submission():
  """
  A helper class for bulk uploads. It parses the given chunk of bulk
  upload and tries to submit them to PyKata
  """
  def __init__(self, bulk_text):
    # initialize everything as default/blank as some things might not be
    # present in submission
    self.data_dict = dict()
    self.bulk_text = bulk_text
    self.multiline_fields = ('description', 'examples')
    self.data_dict['user_id'] = ''

  def __getitem__(self, key):
    try:
      return self.data_dict[key]
    except :
      return None

  def __setitem__(self, key, item):
    # mapping to match form submission for code reuse
    if key=='pubTests':
      key = 'tests'
    if key=='privTests':
      key='other_tests'
    if key=='userID':
      key='user_id'

    self.data_dict[key] = item

  def __str__(self):
    return str(self.bulk_text)

  def parse(self):
    """ Parses a problem submission into data fillable in the submission
    form
    """
    chunks = self.bulk_text.split('#-')
    for chunk in chunks[1:]:
      # ^^ ignores the first comment line of the problem
      chunk_key, chunk_data = chunk.split(':',1)
      chunk_key = chunk_key.strip()
      if chunk_key in self.multiline_fields:
        # these are data surrounded in triple quotes, single or double
        chunk_data = eval(chunk_data).strip('\r\n')
      else:
        chunk_data = chunk_data.strip().strip('\r\n')
      self[chunk_key] = chunk_data

  def talk(self):
    """ Just a testing function, a submission talks about itself to everyone
    """
    print "Hi I am a submission problem on PyKata,"
    print "Let me tell you something about myself - "
    print "I was created by -", self['userID']
    print "My name is -", self['name']
    print "I belong to these categories -", self['categories']
    print "Here, something more about myself --"
    print self['description']
    print "---------------------------------------"
    print "My solution --"
    print self['solution']
    print "----------------------------------------"
    print "My skeleton --"
    print self['skeleton']
    print "----------------------------------------"
    print "Public Tests --"
    print self['pubTests']
    print "----------------------------------------"
    print "Private Tests --"
    print self['privTests']
    print "----------------------------------------"
    print "\n So tell me something about you too?"


def test_submission_parse():
  import urllib2
#  example_file = urllib2.urlopen('http://ideamonk.in/code/foo.py').read()
  example_file = open('/home/amit/foo2.py').read()
  example_submissions = example_file.split('#---')[1:-1]
  # ^^ ignore the first null string and last __main__ test code
  print example_submissions  
  example_submissions = [ Submission(x) for x in example_submissions ]
  
  for key, a_submission in enumerate(example_submissions):
    print "Problem #", key, "parsed as --"
    print "*************************************************************"
    a_submission.parse()
    a_submission.talk()

if __name__=='__main__':
  # testing
  test_submission_parse()