gyb_stdlib_support.py

# ===--- gyb_stdlib_support.py -----------------------*- coding: utf-8 -*-===//
#
# This source file is part of the Swift.org open source project
#
# Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
# Licensed under Apache License v2.0 with Runtime Library Exception
#
# See https://swift.org/LICENSE.txt for license information
# See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors

TRAVERSALS= ['Forward', 'Bidirectional', 'RandomAccess']


defcollectionForTraversal(traversal): # noqa (N802 function name should be lowercase)
iftraversal=='Forward':
return'Collection'
eliftraversal=='Bidirectional':
return'BidirectionalCollection'
eliftraversal=='RandomAccess':
return'RandomAccessCollection'
else:
raiseValueError("Unknown traversal %r"%traversal)


defcollectionTypeName(traversal, mutable, rangeReplaceable): # noqa (N802)
name=collectionForTraversal(traversal)
ifrangeReplaceable:
name='RangeReplaceable'+name
ifmutable:
name='Mutable'+name
returnname


defprotocolsForCollectionFeatures(traversal, mutable, rangeReplaceable): # noqa (N802)
protocols= [collectionForTraversal(traversal)]
ifmutable:
protocols.append('MutableCollection')
ifrangeReplaceable:
protocols.append('RangeReplaceableCollection')
returnprotocols

defdocumentationNameForTraversal(traversal): # noqa (N802)
iftraversal=='Forward':
return'collection'
eliftraversal=='Bidirectional':
return'bidirectional collection'
eliftraversal=='RandomAccess':
return'random-access collection'
else:
raiseValueError("Unknown traversal %r"%traversal)