Accidental refactor, split out legacy.py into separate sumodules and update all call sites.

endpoints/api/search.py

@@ -3,12 +3,12 @@
 from endpoints.api import (ApiResource, parse_args, query_param, truthy_bool, nickname, resource,
                            require_scope, path_param)
 from data import model
-from auth.permissions import (OrganizationMemberPermission, ViewTeamPermission,
-                              ReadRepositoryPermission, UserAdminPermission,
-                              AdministerOrganizationPermission, ReadRepositoryPermission)
+from auth.permissions import (OrganizationMemberPermission, ReadRepositoryPermission,
+                              UserAdminPermission, AdministerOrganizationPermission,
+                              ReadRepositoryPermission)
 from auth.auth_context import get_authenticated_user
 from auth import scopes
-from app import avatar, get_app_url
+from app import avatar
 from operator import itemgetter
 from stringscore import liquidmetal
 from util.names import parse_robot_username
@@ -35,7 +35,7 @@ class EntitySearch(ApiResource):
     organization = None
 
     try:
-      organization = model.get_organization(namespace_name)
+      organization = model.organization.get_organization(namespace_name)
 
       # namespace name was an org
       permission = OrganizationMemberPermission(namespace_name)
@@ -43,7 +43,7 @@ class EntitySearch(ApiResource):
         robot_namespace = namespace_name
 
         if args['includeTeams']:
-          teams = model.get_matching_teams(prefix, organization)
+          teams = model.team.get_matching_teams(prefix, organization)
 
         if args['includeOrgs'] and AdministerOrganizationPermission(namespace_name) \
            and namespace_name.startswith(prefix):
@@ -54,7 +54,7 @@ class EntitySearch(ApiResource):
             'avatar': avatar.get_data_for_org(organization),
           }]
 
-    except model.InvalidOrganizationException:
+    except model.organization.InvalidOrganizationException:
       # namespace name was a user
       user = get_authenticated_user()
       if user and user.username == namespace_name:
@@ -63,7 +63,7 @@ class EntitySearch(ApiResource):
         if admin_permission.can():
           robot_namespace = namespace_name
 
-    users = model.get_matching_users(prefix, robot_namespace, organization)
+    users = model.user.get_matching_users(prefix, robot_namespace, organization)
 
     def entity_team_view(team):
       result = {
@@ -95,18 +95,6 @@ class EntitySearch(ApiResource):
     }
 
 
-  def team_view(orgname, team):
-    view_permission = ViewTeamPermission(orgname, team.name)
-    role = model.get_team_org_role(team).name
-    return {
-      'id': team.id,
-      'name': team.name,
-      'description': team.description,
-      'can_view': view_permission.can(),
-      'role': role
-    }
-
-
 @resource('/v1/find/repository')
 class FindRepositories(ApiResource):
   """ Resource for finding repositories. """
@@ -130,7 +118,7 @@ class FindRepositories(ApiResource):
     if user is not None:
       username = user.username
 
-    matching = model.get_matching_repositories(prefix, username)
+    matching = model.repository.get_matching_repositories(prefix, username)
     return {
       'repositories': [repo_view(repo) for repo in matching
                        if (repo.visibility.name == 'public' or
@@ -174,7 +162,7 @@ def search_entity_view(username, entity, get_short_name=None):
 
 def conduct_team_search(username, query, encountered_teams, results):
   """ Finds the matching teams where the user is a member. """
-  matching_teams = model.get_matching_user_teams(query, get_authenticated_user(), limit=5)
+  matching_teams = model.team.get_matching_user_teams(query, get_authenticated_user(), limit=5)
   for team in matching_teams:
     if team.id in encountered_teams:
       continue
@@ -193,7 +181,7 @@ def conduct_team_search(username, query, encountered_teams, results):
 
 def conduct_admined_team_search(username, query, encountered_teams, results):
   """ Finds matching teams in orgs admined by the user. """
-  matching_teams = model.get_matching_admined_teams(query, get_authenticated_user(), limit=5)
+  matching_teams = model.team.get_matching_admined_teams(query, get_authenticated_user(), limit=5)
   for team in matching_teams:
     if team.id in encountered_teams:
       continue
@@ -212,14 +200,15 @@ def conduct_admined_team_search(username, query, encountered_teams, results):
 
 def conduct_repo_search(username, query, results):
   """ Finds matching repositories. """
-  def can_read(repository):
-    if repository.is_public:
+  def can_read(repo):
+    if repo.is_public:
       return True
 
-    return ReadRepositoryPermission(repository.namespace_user.username, repository.name).can()
+    return ReadRepositoryPermission(repo.namespace_user.username, repo.name).can()
 
   only_public = username is None
-  matching_repos = model.get_sorted_matching_repositories(query, only_public, can_read, limit=5)
+  matching_repos = model.repository.get_sorted_matching_repositories(query, only_public, can_read,
+                                                                     limit=5)
 
   for repo in matching_repos:
     repo_score = math.log(repo.count or 1, 10) or 1
@@ -242,7 +231,7 @@ def conduct_repo_search(username, query, results):
 
 def conduct_namespace_search(username, query, results):
   """ Finds matching users and organizations. """
-  matching_entities = model.get_matching_user_namespaces(query, username, limit=5)
+  matching_entities = model.user.get_matching_user_namespaces(query, username, limit=5)
   for entity in matching_entities:
     results.append(search_entity_view(username, entity))
 
@@ -252,7 +241,7 @@ def conduct_robot_search(username, query, results):
   def get_short_name(name):
     return parse_robot_username(name)[1]
 
-  matching_robots = model.get_matching_robots(query, username, limit=5)
+  matching_robots = model.user.get_matching_robots(query, username, limit=5)
   for robot in matching_robots:
     results.append(search_entity_view(username, robot, get_short_name))