refactor main, bump version, added a test

manwrapper/main.py

@@ -1,25 +1,27 @@
 import subprocess
 import sys
 
-def main():
- if len(sys.argv) < 2 or sys.argv[1] in ['-h', '--help']:
- print("Usage: manwrapper <topic>")
- print("Please provide a topic for the manual.")
- sys.exit(1)
-
- manTopic = sys.argv[1]
-
- cmd = ['man', '-P', 'cat', manTopic]
-
+def get_man_page(topic):
+ cmd = ['man', '-P', 'cat', topic]
  p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  stdout, stderr = p.communicate()
 
  encoding = sys.getdefaultencoding()
 
  if stdout:
- print(stdout.decode(encoding))
+ return stdout.decode(encoding)
  if stderr:
- print("Error:", stderr.decode(encoding))
+ return "Error: " + stderr.decode(encoding)
+
+def main():
+ if len(sys.argv) < 2 or sys.argv[1] in ['-h', '--help']:
+ print("Usage: manwrapper <topic>")
+ print("Please provide a topic for the manual.")
+ sys.exit(1)
+
+ manTopic = sys.argv[1]
+ output = get_man_page(manTopic)
+ print(output)
 
 if __name__ == "__main__":
  main()

setup.py

@@ -2,7 +2,7 @@
 
 setup(
  name="manwrapper",
- version="0.2",
+ version="0.4",
  description="A simple command-line tool to view man pages.",
  long_description=open('README.md').read(),
  long_description_content_type="text/markdown",

tests/test_manwrapper.py

@@ -0,0 +1,12 @@
+# test_manwrapper.py
+
+from manwrapper.main import get_man_page
+
+def test_get_man_page():
+ # Test a known command
+ result = get_man_page('ls')
+ assert "list directory contents" in result
+
+ # Test an unknown command
+ result = get_man_page('nonexistentcommand')
+ assert "Error:" in result