#!/usr/bin/python2.5

#

# Copyright 2008 the Melange authors.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#   http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"""Seeds or clears the datastore.

"""

__authors__ = [

    '"Sverre Rabbelier" <sverre@rabbelier.nl>',

    '"James Levy" <jamesalexanderlevy@gmail.com>',

  ]

import itertools

import logging

import random

import datetime

from google.appengine.api import users

from google.appengine.api import memcache

from google.appengine.ext import db

from django import http

from soc.logic.models.document import logic as document_logic

from soc.logic.models.ranker_root import logic as ranker_root_logic

from soc.logic.models.survey import logic as survey_logic

from soc.logic.models.user import logic as user_logic

from soc.logic import accounts

from soc.logic import dicts

from soc.models import student_proposal

from soc.models.document import Document

from soc.models.host import Host

from soc.models.mentor import Mentor

from soc.models.notification import Notification

from soc.models.org_admin import OrgAdmin

from soc.models.organization import Organization

from soc.models.org_app import OrgApplication

from soc.models.program import Program

from soc.models.ranker_root import RankerRoot

from soc.models.site import Site

from soc.models.student import Student

from soc.models.student_project import StudentProject

from soc.models.student_proposal import StudentProposal

from soc.models.survey import Survey, SurveyContent

from soc.models.survey_record import SurveyRecord

from soc.models.sponsor import Sponsor

from soc.models.timeline import Timeline

from soc.models.user import User

class Error(Exception):

  """Base class for all exceptions raised by this module.

  """

  pass

def ensureUser():

  """Returns the current user account and associated user object.

  """

  account = accounts.getCurrentAccount()

  if not account:

    account = users.User(email='test@example.com')

  user_properties = {

      'key_name': 'test',

      'link_id': 'test',

      'account': account,

      'name': 'Test',

      }

  current_user = User(**user_properties)

  current_user.put()

  return account, current_user

class Seeder(object):

  """A Seeder can seed Melange types.

  """

  def type(self):

    """Returns the type to be seeded."""

    raise NotImplementedError

  def seed(self, i, entities=None, **kwargs):

    """Seed the ith instance of this type.

    Args:

      i, int: which to seed

      entities, list of type()'s: if None, persist at the end of this call.

        if non-None, append the created entity to entities instead of

        persisting.

      kwargs: the dictionary returned by commonSeedArgs

    """

    raise NotImplementedError

  def commonSeedArgs(self, request):

    """Find common information for seeding that's common across entities

    seeded in one request.

    Returns:

      dictionary of str->value; passed to each call of seed() for this

      request

    """

    raise NotImplementedError

class UserSeeder(Seeder):

  """A Seeder for Melange User model.

  """

  def type(self):

    return User

  # pylint: disable-msg=W0221

  def seed(self, i, entities=None):

    user_properties = {

        'key_name': 'user_%04d' % i,

        'link_id': 'user_%04d' % i,

        'account': users.User(email='user_%04d@example.com' % i),

        'name': 'User %04d' % i,

        }

    entity = User(**user_properties)

    if entities is None:

      entity.put()

      user_logic._onCreate(entity)

    else:

      entities.append(entity)

  def commonSeedArgs(self, request):

    return {}

class OrganizationSeeder(Seeder):

  """A Seeder for Melange Organization model.

  """

  def type(self):

    return Organization

  # pylint: disable-msg=W0221

  def seed(self, i, entities=None, current_user=None, gsoc2009=None):

    properties = {

        'key_name': 'google/gsoc2009/org_%04d' % i,

        'link_id': 'org_%04d' % i,

        'name': 'Organization %04d' % i,

        'short_name': 'Org %04d' % i,

        'founder': current_user,

        'scope_path': 'google/gsoc2009',

        'scope': gsoc2009,

        'status': 'active',

        'email': 'org_%04d@example.com' % i,

        'home_page': 'http://code.google.com/p/soc',

        'description': 'Melange, share the love!',

        'license_name': 'Apache License',

        'contact_street': 'Some Street',

        'contact_city': 'Some City',

        'contact_country': 'United States',

        'contact_postalcode': '12345',

        'phone': '1-555-BANANA',

        'ideas': 'http://code.google.com/p/soc/issues',

        }

    org = Organization(**properties)

    if entities is None:

      org.put()

    else:

      entities.append(org)

  def commonSeedArgs(self, request):

    _, current_user = ensureUser()

    gsoc2009 = Program.get_by_key_name('google/gsoc2009')

    if not gsoc2009:

      raise Error('Run seed_db first')

    return dict(current_user=current_user,

                gsoc2009=gsoc2009)

class OrgApplicationSeeder(Seeder):

  """A Seeder for Melange OrgApplication model.

  """

  def type(self):

    return OrgApplication

  def commonSeedArgs(self, request):

    _, current_user = ensureUser()

    gsoc2009 = Program.get_by_key_name('google/gsoc2009')

    if not gsoc2009:

      raise Error('Run seed_db first')

    status = request.GET.get('status', 'pre-accepted')

    return dict(current_user=current_user,

                gsoc2009=gsoc2009,

                status=status)

  # pylint: disable-msg=W0221

  def seed(self, i, entities=None, current_user=None, gsoc2009=None,

           status=None):

    properties = {

        'key_name': 'google/gsoc2009/org_%04d' % i,

        'link_id': 'org_%04d' % i,

        'name': 'Org App %04d' % i,

        'scope_path': 'google/gsoc2009',

        'scope': gsoc2009,

        'status': status,

        'applicant': current_user,

        'home_page': 'http://www.google.com',

        'email': 'org@example.com',

        'irc_channel': '#care',

        'pub_mailing_list': 'http://groups.google.com',

        'dev_mailing_list': 'http://groups.google.com',

        'description': 'This is an awesome org!',

        'why_applying': 'Because we can',

        'member_criteria': 'They need to be awesome',

        'license_name': 'Apache License, 2.0',

        'ideas': 'http://code.google.com/p/soc/issues',

        'contrib_disappears': 'We use google to find them',

        'member_disappears': 'See above',

        'encourage_contribs': 'We offer them cookies.',

        'continued_contribs': 'We promise them a cake.',

        'agreed_to_admin_agreement': True,

        }

    org_application = OrgApplication(**properties)

    if entities is None:

      org_application.put()

    else:

      entities.append(org_application)

def seed(request, *args, **kwargs):

  """Seeds the datastore with some default values.

  """

  site_properties = {

      'key_name': 'site',

      'link_id': 'site',

      }

  site = Site(**site_properties)

  site.put()

  _, current_user = ensureUser()

  seeder = UserSeeder()

  for i in range(15):

    seeder.seed(i)

  group_properties = {

       'key_name': 'google',

       'link_id': 'google',

       'name': 'Google Inc.',

       'short_name': 'Google',

       'founder': current_user,

       'home_page': 'http://www.google.com',

       'email': 'ospo@google.com',

       'description': 'This is the profile for Google.',

       'contact_street': 'Some Street',

       'contact_city': 'Some City',

       'contact_country': 'United States',

       'contact_postalcode': '12345',

       'phone': '1-555-BANANA',

       'status': 'active',

       }

  google = Sponsor(**group_properties)

  google.put()

  role_properties = {

      'key_name': 'google/test',

      'link_id': 'test',

      'scope': google,

      'scope_path': 'google',

      'user': current_user,

      'given_name': 'Test',

      'surname': 'Example',

      'name_on_documents': 'Test Example',

      'email': 'test@example.com',

      'res_street': 'Some Street',

      'res_city': 'Some City',

      'res_state': 'Some State',

      'res_country': 'United States',

      'res_postalcode': '12345',

      'phone': '1-555-BANANA',

      'birth_date': db.DateProperty.now(),

      'agreed_to_tos': True,

      }

  google_host = Host(**role_properties)

  google_host.put()

  timeline_properties = {

        'key_name': 'google/gsoc2009',

        'link_id': 'gsoc2009',

        'scope_path': 'google',

        'scope': google,

        }

  gsoc2009_timeline = Timeline(**timeline_properties)

  gsoc2009_timeline.put()

  program_properties = {

      'key_name': 'google/gsoc2009',

      'link_id': 'gsoc2009',

      'scope_path': 'google',

      'scope': google,

      'name': 'Google Summer of Code 2009',

      'short_name': 'GSoC 2009',

      'group_label': 'GSOC',

      'description': 'This is the program for GSoC 2009.',

      'apps_tasks_limit': 42,

      'slots': 42,

      'timeline': gsoc2009_timeline,

      'status': 'visible',

      }

  gsoc2009 = Program(**program_properties)

  gsoc2009.put()

  # TODO: Use real GHOPProgram here

  timeline_properties = {

        'key_name': 'google/ghop2009',

        'link_id': 'ghop2009',

        'scope_path': 'google',

        'scope': google,

        }

  ghop2009_timeline = Timeline(**timeline_properties)

  ghop2009_timeline.put()

  program_properties.update({

      'key_name': 'google/ghop2009',

      'link_id': 'ghop2009',

      'name': 'Google Highly Open Participation Contest 2009',

      'short_name': 'GHOP 2009',

      'group_label': 'GHOP',

      'description': 'This is the program for GHOP 2009.',

      'timeline': ghop2009_timeline,

      })

  ghop2009 = Program(**program_properties)

  ghop2009.put()

  org_app_properties = {

    'scope_path': 'google/gsoc2009',

    'scope': gsoc2009,

    'applicant': current_user,

    'home_page': 'http://www.google.com',

    'email': 'org@example.com',

    'irc_channel': '#care',

    'pub_mailing_list': 'http://groups.google.com',

    'dev_mailing_list': 'http://groups.google.com',

    'description': 'This is an awesome org!',

    'why_applying': 'Because we can',

    'member_criteria': 'They need to be awesome',

    'status': 'pre-accepted',

    'license_name': 'Apache License, 2.0',

    'ideas': 'http://code.google.com/p/soc/issues',

    'contrib_disappears': 'We use google to find them',

    'member_disappears': 'See above',

    'encourage_contribs': 'We offer them cookies.',

    'continued_contribs': 'We promise them a cake.',

    'agreed_to_admin_agreement': True,

    }

  for i in range(10):

    org_app_properties['key_name'] = 'google/gsoc2009/org_%04d' % i

    org_app_properties['link_id'] = 'org_%04d' % i

    org_app_properties['name'] = 'Org App %04d' % i

    entity = OrgApplication(**org_app_properties)

    entity.put()

  org_app_properties['status'] = 'pre-rejected'

  for i in range(10, 20):

    org_app_properties['key_name'] = 'google/gsoc2009/loser_%d' % i

    org_app_properties['link_id'] = 'loser_%d' % i

    org_app_properties['name'] = 'Loser %d' % i

    entity = OrgApplication(**org_app_properties)

    entity.put()

  group_properties.update({

    'key_name': 'google/ghop2009/melange',

    'link_id': 'melange',

    'name': 'Melange Development Team',

    'short_name': 'Melange',

    'scope_path': 'google/ghop2009',

    'scope': ghop2009,

    'home_page': 'http://code.google.com/p/soc',

    'description': 'Melange, share the love!',

    'license_name': 'Apache License',

    'ideas': 'http://code.google.com/p/soc/issues',

    })

  melange = Organization(**group_properties)

  melange.put()

  # create a new ranker

  ranker_root_logic.create(student_proposal.DEF_RANKER_NAME, melange,

      student_proposal.DEF_SCORE, 100)

  group_properties.update({

    'scope_path': 'google/gsoc2009',

    'scope': gsoc2009,

    })

  orgs = []

  for i in range(15):

    group_properties.update({

        'key_name': 'google/gsoc2009/org_%d' % i,

        'link_id': 'org_%d' % i,

        'name': 'Organization %d' % i,

        'short_name': 'Org %d' % i,

        'description': 'Organization %d!' % i,

        })

    entity = Organization(**group_properties)

    orgs.append(entity)

    entity.put()

    # create a new ranker

    ranker_root_logic.create(student_proposal.DEF_RANKER_NAME, entity,

        student_proposal.DEF_SCORE, 100)

    if i < 2:

      role_properties.update({

          'key_name': 'google/gsoc2009/org_%d/test' % i,

          'link_id': 'test',

          'scope_path': 'google/gsoc2009/org_%d' % i,

          'scope': entity,

          'program': gsoc2009,

          })

      # Admin for the first org

      if i == 0:

        org_1_admin = OrgAdmin(**role_properties)

        org_1_admin.put()

      # Only a mentor for the second org

      if i == 1:

        org_1_admin = OrgAdmin(**role_properties)

        org_1_admin.put()

        org_1_mentor = Mentor(**role_properties)

        org_1_mentor.put()

  role_properties.update({

      'key_name': 'google/ghop2009/melange/test',

      'link_id': 'test',

      'scope_path': 'google/ghop2009/melange',

      'scope': melange,

      'program': ghop2009,

      })

  melange_admin = OrgAdmin(**role_properties)

  melange_admin.put()

  melange_mentor = Mentor(**role_properties)

  melange_mentor.put()

  student_id = 'test'

  student_properties = {

      'key_name': gsoc2009.key().name() + "/" + student_id,

      'link_id': student_id,