Port logs and robots.

endpoints/api/logs.py

@@ -0,0 +1,121 @@
+import json
+
+from datetime import datetime, timedelta
+from flask.ext.restful import abort
+
+from endpoints.api import (resource, nickname, ApiResource, query_param, parse_args,
+                           RepositoryParamResource, require_repo_admin)
+from auth.permissions import AdministerOrganizationPermission, AdministerOrganizationPermission
+from auth.auth_context import get_authenticated_user
+from data import model
+
+
+def log_view(log):
+  view = {
+    'kind': log.kind.name,
+    'metadata': json.loads(log.metadata_json),
+    'ip': log.ip,      
+    'datetime': log.datetime,
+  }
+
+  if log.performer:
+    view['performer'] = {
+      'kind': 'user',
+      'name': log.performer.username,
+      'is_robot': log.performer.robot,
+    }
+
+  return view
+
+
+def get_logs(namespace, start_time, end_time, performer_name=None,
+             repository=None):
+  performer = None
+  if performer_name:
+    performer = model.get_user(performer_name)
+
+  if start_time:
+    try:
+      start_time = datetime.strptime(start_time + ' UTC', '%m/%d/%Y %Z')
+    except ValueError:
+      start_time = None
+
+  if not start_time:  
+    start_time = datetime.today() - timedelta(7) # One week
+
+  if end_time:
+    try:
+      end_time = datetime.strptime(end_time + ' UTC', '%m/%d/%Y %Z')
+      end_time = end_time + timedelta(days=1)
+    except ValueError:
+      end_time = None
+      
+  if not end_time:  
+    end_time = datetime.today()
+
+  logs = model.list_logs(namespace, start_time, end_time, performer=performer,
+                         repository=repository)
+  return {
+    'start_time': start_time,
+    'end_time': end_time,
+    'logs': [log_view(log) for log in logs]
+  }
+
+
+@resource('/v1/repository/<path:repository>/logs')
+class RepositoryLogs(RepositoryParamResource):
+  """ Resource for fetching logs for the specific repository. """
+  @require_repo_admin
+  @nickname('listRepoLogs')
+  @parse_args
+  @query_param('starttime', 'Earliest time from which to get logs (%m/%d/%Y %Z)', type=str)
+  @query_param('endtime', 'Latest time to which to get logs (%m/%d/%Y %Z)', type=str)
+  def get(self, args, namespace, repository):
+    """ List the logs for the specified repository. """
+    repo = model.get_repository(namespace, repository)
+    if not repo:
+      abort(404)
+
+    start_time = args['starttime']
+    end_time = args['endtime']
+    return get_logs(namespace, start_time, end_time, repository=repo)
+
+
+@resource('/v1/organization/<orgname>/logs')
+class OrgLogs(ApiResource):
+  """ Resource for fetching logs for the entire organization. """
+  @nickname('listOrgLogs')
+  @parse_args
+  @query_param('starttime', 'Earliest time from which to get logs. (%m/%d/%Y %Z)', type=str)
+  @query_param('endtime', 'Latest time to which to get logs. (%m/%d/%Y %Z)', type=str)
+  @query_param('performer', 'Username for which to filter logs.', type=str)
+  # @org_api_call('list_user_logs')
+  def get(self, args, orgname):
+    """ List the logs for the specified organization. """
+    permission = AdministerOrganizationPermission(orgname)
+    if permission.can():
+      performer_name = args['performer']
+      start_time = args['starttime']
+      end_time = args['endtime']
+
+      return get_logs(orgname, start_time, end_time, performer_name=performer_name)
+
+    abort(403)
+
+
+@resource('/v1/user/logs')
+class UserLogs(ApiResource):
+  """ Resource for fetching logs for the current user. """
+  @nickname('listUserLogs')
+  @parse_args
+  @query_param('starttime', 'Earliest time from which to get logs. (%m/%d/%Y %Z)', type=str)
+  @query_param('endtime', 'Latest time to which to get logs. (%m/%d/%Y %Z)', type=str)
+  @query_param('performer', 'Username for which to filter logs.', type=str)
+  def get(self, args):
+    """ List the logs for the current user. """
+    performer_name = args['performer']
+    start_time = args['starttime']
+    end_time = args['endtime']
+
+    return get_logs(get_authenticated_user().username, start_time, end_time,
+                    performer_name=performer_name)