Skip to content

Commit

Permalink
Make the position_statements module importable
Browse files Browse the repository at this point in the history
  • Loading branch information
@mhl
mhl committed Oct 3, 2017
1 parent 4671ee9 commit 7a4beaf
Showing 1 changed file with 92 additions and 91 deletions.
183 changes: 92 additions & 91 deletions position_statements.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -72,105 +72,106 @@ def expanded_datavalue(datavalue):
time.precision = datavalue['value']['precision']
return time

if __name__ == '__main__':

if len(sys.argv) == 1:
sys.exit("Usage: %s <filename> [<user_for_attribution>]" % sys.argv[0])
if len(sys.argv) == 1:
sys.exit("Usage: %s <filename> [<user_for_attribution>]" % sys.argv[0])

p = Path(sys.argv[1])
p = Path(sys.argv[1])

if not p.is_file():
sys.exit('"%s" is not a file' % sys.argv[1])
if not p.is_file():
sys.exit('"%s" is not a file' % sys.argv[1])

try:
user_name = sys.argv[2]
except IndexError:
user_name = None

site = pywikibot.Site()
repo = site.data_repository()

with p.open() as f:
statements_string = f.read()

statements = [s for s in statements_string.split("\n") if s.strip() != '']

commands = []

# Parse statements
for statement in statements:
command = {}
parts = [s.strip() for s in statement.split("\t")]
command['item'] = parts[0]
command['property'] = parts[1]
command['datavalue'] = parse_value(parts[2])
qualifiers = parts[3:]
qualifier_pairs = list(zip(qualifiers[::2], qualifiers[1::2]))
command['qualifiers'] = []
command['sources'] = []
for p, v in qualifier_pairs:
what = 'sources' if p.startswith('S') else 'qualifiers'
prop = re.sub(r'^S', 'P', p)
q = {'property': prop, 'datavalue': parse_value(v)}
command[what].append(q)

# Validate statements
m = item_re.match(command['item'])
if m is None:
sys.exit("Invalid item ID format: {}".format(command['item']))
try:
user_name = sys.argv[2]
except IndexError:
user_name = None

site = pywikibot.Site()
repo = site.data_repository()

with p.open() as f:
statements_string = f.read()

statements = [s for s in statements_string.split("\n") if s.strip() != '']

commands = []

# Parse statements
for statement in statements:
command = {}
parts = [s.strip() for s in statement.split("\t")]
command['item'] = parts[0]
command['property'] = parts[1]
command['datavalue'] = parse_value(parts[2])
qualifiers = parts[3:]
qualifier_pairs = list(zip(qualifiers[::2], qualifiers[1::2]))
command['qualifiers'] = []
command['sources'] = []
for p, v in qualifier_pairs:
what = 'sources' if p.startswith('S') else 'qualifiers'
prop = re.sub(r'^S', 'P', p)
q = {'property': prop, 'datavalue': parse_value(v)}
command[what].append(q)

# Validate statements
m = item_re.match(command['item'])
if m is None:
sys.exit("Invalid item ID format: {}".format(command['item']))

if command['property'] != 'P39':
sys.exit("Only P39 statements are supported currently. (Got {})".format(command['property']))
if command['property'] != 'P39':
sys.exit("Only P39 statements are supported currently. (Got {})".format(command['property']))

if len(qualifiers) % 2 != 0:
sys.exit("Odd number of qualifier/source pairs detected: {}".format(qualifiers))
if len(qualifiers) % 2 != 0:
sys.exit("Odd number of qualifier/source pairs detected: {}".format(qualifiers))

for qualifier in command['qualifiers']:
m = property_re.match(qualifier['property'])
if m is None:
sys.exit("Invalid qualifier property: {}".format(qualifier['property']))
for qualifier in command['qualifiers']:
m = property_re.match(qualifier['property'])
if m is None:
sys.exit("Invalid qualifier property: {}".format(qualifier['property']))

for source in command['sources']:
m = property_re.match(source['property'])
if m is None:
sys.exit("Invalid source property: {}".format(source['property']))
for source in command['sources']:
m = property_re.match(source['property'])
if m is None:
sys.exit("Invalid source property: {}".format(source['property']))

# Validations passed, add to list of commands.
commands.append(command)
# Validations passed, add to list of commands.
commands.append(command)

for command in commands:
print(command)
summary = 'Edited with PositionStatements'
if user_name:
summary += ' on behalf of [[User:{}]]'.format(user_name)
for command in commands:
print(command)
summary = 'Edited with PositionStatements'
if user_name:
summary += ' on behalf of [[User:{}]]'.format(user_name)

# Get the item we want to modify
item = pywikibot.ItemPage(repo, command['item'])
try:
item.get()
except pywikibot.IsRedirectPage:
item = item.getRedirectTarget()

# Get the claim we're dealing with
claim = pywikibot.Claim(site, command['property'])

# Create a new P39 statement pointing to Q123
claim.setTarget(expanded_datavalue(command['datavalue']))

# Add the claim to the item.
item.addClaim(claim, summary=summary)

for qualifier in command['qualifiers']:
qualifier_claim = pywikibot.Claim(
site, qualifier['property'], isQualifier=True
)
qualifier_claim.setTarget(expanded_datavalue(qualifier['datavalue']))
claim.addQualifier(qualifier_claim, summary=summary)

sources = []
for source in command['sources']:
source_claim = pywikibot.Claim(
site, source['property'], isReference=True
)
source_claim.setTarget(expanded_datavalue(source['datavalue']))
sources.append(source_claim)
claim.addSources(sources, summary=summary)
# Get the item we want to modify
item = pywikibot.ItemPage(repo, command['item'])
try:
item.get()
except pywikibot.IsRedirectPage:
item = item.getRedirectTarget()

# Get the claim we're dealing with
claim = pywikibot.Claim(site, command['property'])

# Create a new P39 statement pointing to Q123
claim.setTarget(expanded_datavalue(command['datavalue']))

# Add the claim to the item.
item.addClaim(claim, summary=summary)

for qualifier in command['qualifiers']:
qualifier_claim = pywikibot.Claim(
site, qualifier['property'], isQualifier=True
)
qualifier_claim.setTarget(expanded_datavalue(qualifier['datavalue']))
claim.addQualifier(qualifier_claim, summary=summary)

sources = []
for source in command['sources']:
source_claim = pywikibot.Claim(
site, source['property'], isReference=True
)
source_claim.setTarget(expanded_datavalue(source['datavalue']))
sources.append(source_claim)
claim.addSources(sources, summary=summary)

0 comments on commit 7a4beaf

Please sign in to comment.