import dont_tread_on_memes
import click


@click.command()
@click.option("--message", prompt="Don't _____ me",
              help=("The word or phrase to substitute for 'tread' in 'don't "
                    "tread on me'"))
@click.option("--save", default=None, help="Where to save the image")
def tread(message, save):
    flag = dont_tread_on_memes.tread_on(message)
    if save is not None:
        flag.save(save)
    else:
        flag.show()


if __name__ == "__main__":
    tread()

import dont_tread_on_memes
import click


@click.command()
@click.option("--message", prompt="Don't _____ me",
              help=("The word or phrase to substitute for 'tread' in 'don't "
                    "tread on me'"))
@click.option("--format/--no-format", default=True,
              help=("Use the provided message as the entire caption instead of"
                    " formatting it as 'Don't [message] me'"))
@click.option("--save", default=None, help="Where to save the image")
def tread(message, format, save):
    # Generate the flag
    if format is True:
        flag = dont_tread_on_memes.dont_me(message)
    else:
        flag = dont_tread_on_memes.tread_on(message)

    # Save or show
    if save is not None:
        flag.save(save)
    else:
        flag.show()


if __name__ == "__main__":
    tread()

Allow 'raw' captioning via the --no-format flag
Allow 'raw' captioning via the --no-format flag

  import dont_tread_on_memes
  import click
  
  
  @click.command()
  @click.option("--message", prompt="Don't _____ me",
                help=("The word or phrase to substitute for 'tread' in 'don't "
                      "tread on me'"))
+ @click.option("--format/--no-format", default=True,
+               help=("Use the provided message as the entire caption instead of"
+                     " formatting it as 'Don't [message] me'"))
  @click.option("--save", default=None, help="Where to save the image")
- def tread(message, save):
+ def tread(message, format, save):
+     # Generate the flag
+     if format is True:
+         flag = dont_tread_on_memes.dont_me(message)
+     else:
-     flag = dont_tread_on_memes.tread_on(message)
+         flag = dont_tread_on_memes.tread_on(message)
+ 
+     # Save or show
      if save is not None:
          flag.save(save)
      else:
          flag.show()
  
  
  if __name__ == "__main__":
      tread()
  
Allow 'raw' captioning via the --no-format flag
## Code Before:
import dont_tread_on_memes
import click


@click.command()
@click.option("--message", prompt="Don't _____ me",
              help=("The word or phrase to substitute for 'tread' in 'don't "
                    "tread on me'"))
@click.option("--save", default=None, help="Where to save the image")
def tread(message, save):
    flag = dont_tread_on_memes.tread_on(message)
    if save is not None:
        flag.save(save)
    else:
        flag.show()


if __name__ == "__main__":
    tread()

## Instruction:
Allow 'raw' captioning via the --no-format flag
## Code After:
import dont_tread_on_memes
import click


@click.command()
@click.option("--message", prompt="Don't _____ me",
              help=("The word or phrase to substitute for 'tread' in 'don't "
                    "tread on me'"))
@click.option("--format/--no-format", default=True,
              help=("Use the provided message as the entire caption instead of"
                    " formatting it as 'Don't [message] me'"))
@click.option("--save", default=None, help="Where to save the image")
def tread(message, format, save):
    # Generate the flag
    if format is True:
        flag = dont_tread_on_memes.dont_me(message)
    else:
        flag = dont_tread_on_memes.tread_on(message)

    # Save or show
    if save is not None:
        flag.save(save)
    else:
        flag.show()


if __name__ == "__main__":
    tread()
