mirror of
https://github.com/google/nomulus.git
synced 2025-05-01 20:47:52 +02:00
b08e317505
This tool finds all registrars that have a null name, or a null localizedAddress field, and populates them as follows: - The registrar's name is updated to its clientId - The registrar's empty localizedAddress fields are replaced with fake data ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=155739946
73 lines
2.9 KiB
Java
73 lines
2.9 KiB
Java
// Copyright 2017 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// 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.
|
|
|
|
package google.registry.tools.javascrap;
|
|
|
|
import static com.google.common.base.MoreObjects.firstNonNull;
|
|
import static google.registry.model.common.EntityGroupRoot.getCrossTldKey;
|
|
import static google.registry.model.ofy.ObjectifyService.ofy;
|
|
|
|
import com.beust.jcommander.Parameters;
|
|
import com.google.common.collect.ImmutableList;
|
|
import google.registry.model.registrar.Registrar;
|
|
import google.registry.model.registrar.Registrar.Builder;
|
|
import google.registry.model.registrar.RegistrarAddress;
|
|
import google.registry.tools.MutatingCommand;
|
|
import java.util.Objects;
|
|
|
|
/**
|
|
* Scrap tool to update Registrars with null registrarName or localizedAddress fields.
|
|
*
|
|
* <p>This sets a null registrarName to the key name, and null localizedAddress fields to fake data.
|
|
*/
|
|
@Parameters(
|
|
separators = " =",
|
|
commandDescription = "Populate previously null required registrar fields."
|
|
)
|
|
public class PopulateNullRegistrarFieldsCommand extends MutatingCommand {
|
|
|
|
@Override
|
|
protected void init() throws Exception {
|
|
for (Registrar registrar : ofy().load().type(Registrar.class).ancestor(getCrossTldKey())) {
|
|
Builder changeBuilder = registrar.asBuilder();
|
|
changeBuilder.setRegistrarName(
|
|
firstNonNull(registrar.getRegistrarName(), registrar.getClientId()));
|
|
|
|
RegistrarAddress address = registrar.getLocalizedAddress();
|
|
if (address == null) {
|
|
changeBuilder.setLocalizedAddress(
|
|
new RegistrarAddress.Builder()
|
|
.setCity("Fakington")
|
|
.setCountryCode("US")
|
|
.setState("FL")
|
|
.setZip("12345")
|
|
.setStreet(ImmutableList.of("123 Fake Street"))
|
|
.build());
|
|
} else {
|
|
changeBuilder.setLocalizedAddress(
|
|
new RegistrarAddress.Builder()
|
|
.setCity(firstNonNull(address.getCity(), "Fakington"))
|
|
.setCountryCode(firstNonNull(address.getCountryCode(), "US"))
|
|
.setState(firstNonNull(address.getState(), "FL"))
|
|
.setZip(firstNonNull(address.getZip(), "12345"))
|
|
.setStreet(firstNonNull(address.getStreet(), ImmutableList.of("123 Fake Street")))
|
|
.build());
|
|
}
|
|
Registrar changedRegistrar = changeBuilder.build();
|
|
if (!Objects.equals(registrar, changedRegistrar)) {
|
|
stageEntityChange(registrar, changedRegistrar);
|
|
}
|
|
}
|
|
}
|
|
}
|