FilesExpand file tree

 parallel-tests

/

admin.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

100 lines (71 loc) · 2.72 KB

 parallel-tests

/

admin.py

Copy path

Top

File metadata and controls

• Code
• Blame

100 lines (71 loc) · 2.72 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

# Copyright 2016 Google, Inc.

# 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.

# [START datastore_admin_client_create]

from google.cloud.datastore_admin_v1 import DatastoreAdminClient

def client_create():

"""Creates a new Datastore admin client."""

client = DatastoreAdminClient()

print("Admin client created\n")

return client

# [END datastore_admin_client_create]

# [START datastore_admin_entities_export]

def export_entities(project_id, output_url_prefix):

"""

Exports a copy of all or a subset of entities from

Datastore to another storage system, such as Cloud Storage.

"""

# project_id = "project-id"

# output_url_prefix = "gs://bucket-name"

client = DatastoreAdminClient()

op = client.export_entities(

{"project_id": project_id, "output_url_prefix": output_url_prefix}

)

response = op.result(timeout=300)

print("Entities were exported\n")

return response

# [END datastore_admin_entities_export]

# [START datastore_admin_entities_import]

def import_entities(project_id, input_url):

"""Imports entities into Datastore."""

# project_id := "project-id"

# input_url := "gs://bucket-name/overall-export-metadata-file"

client = DatastoreAdminClient()

op = client.import_entities(

{"project_id": project_id, "input_url": input_url}

)

response = op.result(timeout=300)

print("Entities were imported\n")

return response

# [END datastore_admin_entities_import]

# [START datastore_admin_index_get]

def get_index(project_id, index_id):

"""Gets an index."""

# project_id := "my-project-id"

# index_id := "my-index"

client = DatastoreAdminClient()

index = client.get_index({"project_id": project_id, "index_id": index_id})

print("Got index: %v\n", index.index_id)

return index

# [END datastore_admin_index_get]

# [START datastore_admin_index_list]

def list_indexes(project_id):

"""Lists the indexes."""

# project_id := "my-project-id"

client = DatastoreAdminClient()

indexes = []

for index in client.list_indexes({"project_id": project_id}):

indexes.append(index)

print("Got index: %v\n", index.index_id)

print("Got list of indexes\n")

return indexes

# [END datastore_admin_index_list]