Ver Fonte

Set up repo for pip import.

Colby há 4 anos atrás
pai
commit
dec425bc6b

+ 1 - 1
README.md

@@ -12,7 +12,7 @@ Install via pip:
 
 ## Run the script
 
-`python audio_to_visualization.py <arguments>`
+`python audio_to_visualization/audio_to_visualization.py <arguments>`
 
 ### Command line arguments
 

+ 0 - 0
__init__.py


+ 3 - 0
audio_to_visualization/__init__.py

@@ -0,0 +1,3 @@
+from audio_to_visualization.audio_to_visualization import create_visualization
+
+__all__ = "create_visualization"

+ 2 - 2
audio_to_visualization.py → audio_to_visualization/audio_to_visualization.py

@@ -40,11 +40,11 @@ def call_video_creator_with_args():
                       help="opacity for visualization background color (0.0-1.0)", required=False)
 
   args, _ = parser.parse_known_args()
-  create_vizualization(args.audio, args.background, args.output, args.vis_background_to_vid_ratio,
+  create_visualization(args.audio, args.background, args.output, args.vis_background_to_vid_ratio,
                        args.vis_waves_to_vid_ratio, args.vis_color, args.vis_color_opacity,
                        args.background_color, args.background_color_opacity)
 
-def create_vizualization(audio, background, output, vis_background_to_vid_ratio,
+def create_visualization(audio, background, output, vis_background_to_vid_ratio,
                          vis_waves_to_vid_ratio, vis_color, vis_color_opacity,
                          background_color, background_color_opacity):
   # Get metadata for visualization

+ 4 - 4
setup.py

@@ -1,8 +1,8 @@
-#!/usr/bin/env python
+import setuptools
 
-from distutils.core import setup
-
-setup(name='audio_to_visualization',
+setuptools.setup(name='audio_to_visualization',
+      package=setuptools.find_packages(),
+      install_requires=['ffmpeg_python'],
       version='1.0',
       description='Audio to visualization script',
       author='CodingVibe',