Jump to content

User:MilHistBot/AutoStub.cs

From Wikipedia, the free encyclopedia
using System;
using System.Collections.Generic;
using System.IO;
using System.Text.RegularExpressions;
using NDesk.Options;

public class AutoStub
{
    private Bot bot;

    private int unchanged;
    private int upgrade;
    private int newgrade;
    private int downgrade;

    private int  max     = 1;
    private bool debugs  = false;
    private bool force   = false;
    private bool help    = false;
    private bool verbose = false;
    private string filename;

    private string checkpoint ()
    {
        StreamReader streamReader = null;
        string line = null;

        Console.WriteLine ("Checkpoint file is " + filename);
        if (File.Exists (filename))
        {
            try
            {
                streamReader = new StreamReader (filename);
                line = streamReader.ReadLine ();
            }
            catch (Exception e)
            {
                bot.Cred.Warning (e.Message);
            }
            finally
            {
                if (null != streamReader)
                {
                    streamReader.Close();
                }
            }
        }
        Console.WriteLine ("Checkpoint is " + (null == line ? "null" : line));
        return line;
    }

    private void checkpoint (string line)
    {
        StreamWriter streamWriter = null;
        try
        {
            streamWriter = new StreamWriter (filename);
            streamWriter.WriteLine (line);
        }
        catch (Exception e)
        {
            bot.Cred.Warning (e.Message);
        }
        finally
        {
            if (null != streamWriter)
            {
               streamWriter.Close ();
            }
        }
    }

    private bool compareClass (string oldClass, string botClass)
    {
        Dictionary<string, int> classes = new Dictionary<string, int>()
        {
            { "Stub",  0 },
            { "Start", 1 },
            { "List",  1 },
            { "C",     2 },
            { "CL",    2 },
            { "B",     3 },
            { "BL",    3 },
        };

        if (oldClass.Equals (""))
        {
            newgrade++;
            return true;
        }

        if (! classes.ContainsKey (oldClass))
        {
            throw new ApplicationException ("unknown class: '" + oldClass + "'");
        }

        int diff = classes[botClass] - classes[oldClass];
        if (0 > diff)
        {
            downgrade++;
        }
        else if (0 == diff)
        {
            unchanged++;
            return false;
        }
        else if (0 < diff)
        {
            upgrade++;
        }
        return true;
    }

    private void report ()
    {
        bot.Cred.Showtime (String.Format("{0} articles newly rated, {1} downgraded, {2} upgraded, {3} unchanged - total {4}",
                            newgrade, downgrade, upgrade, unchanged, newgrade + downgrade + upgrade + unchanged));
    }

    private void autoCheck (Page article, Page talk)
    {
        try
        {
            bot.Cred.Showtime (article.Title);
            article.Load ();
            talk.Load ();
            var template = talk.MilHist.ProjectTemplate;
            var oldClass = template.Class;
            var rating   = template.Rating;
            var botClass = rating.Class;

            Debug.WriteLine  ("\tOriginal: " + template.Text);
            template.RemoveAll (@"^class$|^importance$|^b\d$|^B-Class-\d$");
            template.Rating = rating;

            Debug.WriteLine ("\told rating = " + oldClass);
            Debug.WriteLine ("\tprediction = " + article.Prediction ());
            Debug.WriteLine ("\tbot rating = " + botClass);

            Debug.WriteLine ("\tModified: " + template.Text);
            var changed = compareClass (oldClass, botClass);
            if (changed && force)
            {
                talk.Save ("AutoStub: Automatic MILHIST checklist reassessment - " + botClass + " class");
                bot.Cred.Showtime ("\tChanged from " + oldClass + " to " + botClass);
            }
        }
        catch (ApplicationException ex)
        {
            string message = "Error in " + article.Title + ": " + ex.Message;
            bot.Cred.Showtime (message);
        }
        catch (Exception ex)
        {
            string message = "Error in " + article.Title + ":\n" + ex.Message + "\n" + ex.StackTrace;
            if (debugs)
            {
                Debug.WriteLine (message);
            }
            else
            {
                bot.Cred.Warning (message);
                bot.Close ();
            }
            Environment.Exit (1);
        }
    }

    private void autoCheck (List<Page> talkPages)
    {
        foreach (var talkPage in talkPages)
        {
            // Could be a subcategory
            if (talkPage.Namespace.Equals ("Talk"))
            {
                autoCheck (talkPage.Article, talkPage);
            }
        }
        report ();
    }

    private static void showHelp (OptionSet options)
    {
        Console.WriteLine ("Usage: mono AutoStub [OPTIONS]+ <article>");
        Console.WriteLine ("Assess article and update the MilHist template on the talk page.");
        Console.WriteLine ();
        Console.WriteLine ("Options:");
        options.WriteOptionDescriptions (Console.Out);
        Environment.Exit (0);
    }

    List<string> options (string [] args)
    {
        var optionSet = new OptionSet () {
            { "d|debug",   "debugging",    v => debugs  = v != null },
            { "f|force",   "update page",  v => force   = v != null },
            { "h|?|help",  "display help", v => help    = v != null },
            { "n|max=",    "number of pages to process",  v => int.TryParse (v, out max) },
            { "v|verbose", "vebosity",     v => verbose = v != null },
        };

        List<string> extras = optionSet.Parse (args);

        if (help)
        {
            showHelp (optionSet);
        }

        return extras;
    }

    private AutoStub (string [] args)
    {
        bot = new Bot ();
        Debug.Bot = bot;
        Debug.On  = debugs;
        filename  = Path.Combine (bot.Cred.LogsDirectory, bot.Cred.Job + ".txt");
        var articles = options (args);

        bot.Cred.Showtime ("started");
        if (articles.Count > 0)
        {
            var articlePage = new Page (bot, articles[0]);
            autoCheck (articlePage, articlePage.Talk);
        }
        else
        {
            int batchSize = 100;
            int batch = Math.Min (max, batchSize);
            var query = new Query ("Stub-Class military history articles", batch, checkpoint());
            do
            {
                var talkPages = bot.Category (query);
                checkpoint (query.Start);
                autoCheck (talkPages);
                max -= talkPages.Count;
                if (max < batch)
                {
                    query.Limit = max.ToString ();
                }
            } while (query.Continue && max > 0);

            if (max > 0)
            {
                File.Delete (filename);
            }
        }
         bot.Cred.Showtime ("done");
    }

    static public void Main (string [] args)
    {
        var autoStub = new AutoStub (args);
    }
}