Merge pull request openedx#482 from yangzhenweiq/HH_register

HH register

lms/envs/production.py

@@ -154,6 +154,8 @@
 EMAIL_FILE_PATH = ENV_TOKENS.get('EMAIL_FILE_PATH', None)
 EMAIL_HOST = ENV_TOKENS.get('EMAIL_HOST', 'localhost')  # django default is localhost
 EMAIL_PORT = ENV_TOKENS.get('EMAIL_PORT', 25)  # django default is 25
+EMAIL_ACCESS_LIST = ENV_TOKENS.get('EMAIL_ACCESS_LIST', ["hh.global"])
+EMAIL_ACCESS_DATE = ENV_TOKENS.get('EMAIL_ACCESS_DATE', 365)
 EMAIL_USE_TLS = ENV_TOKENS.get('EMAIL_USE_TLS', False)  # django default is False
 SITE_NAME = ENV_TOKENS['SITE_NAME']
 HTTPS = ENV_TOKENS.get('HTTPS', HTTPS)

openedx/core/djangoapps/user_api/views.py

@@ -1,6 +1,11 @@
 """HTTP end-points for the User API. """
 
+import pytz
+from datetime import datetime
+from dateutil.relativedelta import relativedelta
+
 from django.contrib.auth.models import User
+from django.conf import settings
 from django.core.exceptions import NON_FIELD_ERRORS, PermissionDenied, ValidationError
 from django.db import transaction
 from django.http import HttpResponse, HttpResponseForbidden
@@ -152,6 +157,12 @@ def post(self, request):
 
         try:
             user = create_account_with_params(request, data)
+            if email.split('@')[-1] in settings.EMAIL_ACCESS_LIST:
+                from membership.models import VIPInfo
+                expired_at = datetime.now(pytz.utc) + \
+                relativedelta(days=+int(settings.EMAIL_ACCESS_DATE))
+                VIPInfo.objects.create(user=user, expired_at=expired_at)
+
         except AccountValidationError as err:
             errors = {
                 err.field: [{"user_message": text_type(err)}]